模块'sklearn'没有属性'cross_validation'

Question

我试图将我的数据集拆分为训练和测试数据集,但我收到此错误:

X_train,X_test,Y_train,Y_test = sklearn.cross_validation.train_test_split(X,df1['ENTRIESn_hourly'])

---

AttributeError                            Traceback (most recent call last)
<ipython-input-53-5445dab94861> in <module>()
----> 1 X_train,X_test,Y_train,Y_test = sklearn.cross_validation.train_test_split(X,df1['ENTRIESn_hourly'])

AttributeError: module 'sklearn' has no attribute 'cross_validation'

我怎么处理这个？

Answer 1

sklearn不会自动导入其子包.如果您只通过:导入import sklearn,那么它将无法正常工作.import sklearn.cross_validation改为导入.

Furhter sklearn.cross_validation将在0.20版本中弃用.请sklearn.model_selection.train_test_split改用.

Answer 2

尝试这个：

from sklearn.model_selection import train_test_split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.33, random_state=101)