时间戳

首页 > 解决方案 > 找不到 sklearn.model_selection 模块

python - 找不到 sklearn.model_selection 模块

问题描述

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

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_selectionand linear_model.

我怎么解决这个问题？我已经尝试升级每个模块。

标签： pythonscikit-learn

解决方案

你用的是什么版本的 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

这有帮助吗？

推荐阅读