ImportError:没有名为model_selection的模块

Question

我正在尝试使用train_test_split函数并写:

from sklearn.model_selection import train_test_split

这就是原因

ImportError: No module named model_selection

为什么？以及如何克服？

Answer 1

我猜你有错误的版本scikit-learn,GitHub上描述了类似的情况.之前(之前v0.18),train_test_split位于cross_validation模块中:

from sklearn.cross_validation import train_test_split

但是,现在它在model_selection模块中:

from sklearn.model_selection import train_test_split

所以你需要最新的版本.

要升级到至少版本0.18,请执行以下操作:

pip install -U scikit-learn

(或者pip3,取决于您的Python版本).如果您以不同的方式安装它,请确保使用其他方法进行更新,例如使用Anaconda时.

Answer 2

更新sklearn

conda更新scikit-learn

Answer 3

我导入时遇到了这个问题GridSearchCV.

刚换sklearn.model_selection到sklearn.grid_search.

Answer 4

我在使用Jupyter Notebook时遇到了同样的问题,无论我在Python 3中更新了什么,conda,我都无法进入Jupyter:

import sklearn
print (sklearn.__version__)
0.17.1

显示scikit-learn-0.18.1

最后,我删除了Anaconda3和Jupyter笔记本并重新安装了新的.我得到了它的工作.

Answer 5

你有sklearn吗？如果不是，请执行以下操作：

sudo pip install sklearn

安装sklearn之后

from sklearn.model_selection import train_test_split

工作正常