scikit-learn - 如何修复Python中的 'name ' cross_validation'未定义'错误

Question

我正在尝试运行 XGBClassifier 参数调整，并在这行代码之后收到“'name 'cross_validation' is not Defined”错误:

  kfold_5 =  cross_validation.KFold(n = len(X), shuffle = True, n_folds = numFolds)

也许我没有导入适当的库？

Answer 1

首先，获取您的版本:

import sklearn
sklearn.__version__

在 scikit-learn 0.17 版本之后，cross_validation.KFold 已迁移到 model_selection.KFold。

如果您有 0.17 版本，请使用此:

from sklearn.cross_validation import KFold

kfold_5 = KFold(n= len(X), n_folds = numFolds, shuffle=True)

如果您的版本高于 0.17，请使用此:

from sklearn.model_selection import KFold

kfold_5 = KFold(n_splits = numFolds, shuffle=True)

Documentation for 0.21 version is here