Как исправить 'имя'cross_validation'не определено' ошибка в Python

Я пытаюсь запустить настройку параметров XGBClassifier и получить сообщение об ошибке "имя" cross_validation "не определено" после этой строки кода:

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

Может быть, я не импортировал соответствующую библиотеку?

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)

Документация для версии 0.21 находится здесь

Другие вопросы по тегам