python - 模块 '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 导入。

此外，sklearn.cross_validation 将在 0.20 版中被弃用。请改用 sklearn.model_selection.train_test_split。