python - 找不到 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

这有帮助吗？