cross_validation inside the function are placed inside model_selection
from sklearn.model_selection import train_test_split
cross_validation inside the function are placed inside model_selection
from sklearn.model_selection import train_test_split