RFECV для классификации, дающей KeyError: 'weight' - PullRequest
2 голосов
/ 25 февраля 2020

Рекурсивное удаление функций с перекрестной проверкой (RFECV) не работает, получая KeyError: 'weight'. Как я вижу, он не может рассчитать коэффициенты, поэтому веса отсутствуют. У меня есть оценка как XGBClassifier.

1 Ответ

2 голосов
/ 25 февраля 2020

RFECV не поддерживается с новой версией XGBoost - 1.0.0, поэтому попробуйте запустить ее с более старыми версиями пакета XGBoost.

Пакет удаления:! Pip uninstall xgboost --y

Установите старый пакет:! Pip install xgboost == 0,90 или 0,80

Кроме того, RFECV не будет работать (занимает много времени) с большими наборами данных, а именно: размер столбца> 10000

...