As pointed out by @amit-gupta in the question above, sklearn.cross_validation has been deprecated. The function train_test_split can now be found here:
from sklearn.model_selection import train_test_split
Simply replace the import statement from the question to the one above.
