Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

ImportError: No module named model_selection

I am trying to use train_test_split function and write:

from sklearn.model_selection import train_test_split

and this causes

ImportError: No module named model_selection

Why? And how to overcome?

like image 463
Dims Avatar asked Nov 20 '16 13:11

Dims


2 Answers

I guess you have the wrong version of scikit-learn, a similar situation was described here on GitHub. Previously (before v0.18), train_test_split was located in the cross_validation module:

from sklearn.cross_validation import train_test_split

However, now it's in the model_selection module:

from sklearn.model_selection import train_test_split

so you'll need the newest version.

To upgrade to at least version 0.18, do:

pip install -U scikit-learn

(Or pip3, depending on your version of Python). If you've installed it in a different way, make sure you use another method to update, for example when using Anaconda.

like image 188
linusg Avatar answered Nov 09 '22 08:11

linusg


Update sklearn:

conda update scikit-learn
like image 26
Farhad Maleki Avatar answered Nov 09 '22 07:11

Farhad Maleki