找不到sklearn.model_selection模块

Question

我正在尝试根据数据进行线性回归，但是当我尝试以下操作时：

from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression

它给了我这个错误：

line 4, in <module>
    from sklearn.model_selection import train_test_split
KeyError: 32

sklearn，numpy和scipy都已安装，但是我也无法使用sklearn.model_selection和linear_model。

如何解决此问题？我已经尝试升级每个模块。

Answer 1

您正在使用哪个版本的sklearn？

在版本0.19和更低版本中，功能train_test_split()位于sklearn.cross_validation中。 要使用它：

from sklearn.cross_validation import train_test_split

从0.20版开始，可以在sklearn.model_selection中找到它 要使用它：

from sklearn.model_selection import train_test_split

有帮助吗？