I am trying to use train_test_split function and write:
train_test_split
from sklearn.model_selection import train_test_split
and this causes
do you have sklearn? if not, do the following:
sudo pip install sklearn
After installing sklearn
works fine