python - 导入错误 : No module named model_selection

Question

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

from sklearn.model_selection import train_test_split

这会导致

ImportError: No module named model_selection

为什么？又如何克服？

Answer 1

我猜你的 scikit-learn 版本有误，描述了类似的情况 here on 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 时。