03
Май
2020

xgboost.cv как сделать так, чтобы метки попадали в тестовые выборки?

xgboost.cv(params=params, dtrain=dtrain, num_boost_round=1000, nfold=8)

При большом количестве разбиений, например 8, метки перестают попадать в тестовые выборки, насколько я поняла из ошибки:

Check failed: !auc_error AUC: the dataset only contains pos or neg samples

При nfold=4 Все работает без ошибок.

Помогите найти более правильное и универсальное решение с кроссвалидацией, для будущих данных. Т.к nfold=4 для неизвестных данных может не подойти

Источник: https://ru.stackoverflow.com/questions/1119544/xgboost-cv-%D0%BA%D0%B0%D0%BA-%D1%81%D0%B4%D0%B5%D0%BB%D0%B0%D1%82%D1%8C-%D1%82%D0%B0%D0%BA-%D1%87%D1%82%D0%BE%D0%B1%D1%8B-%D0%BC%D0%B5%D1%82%D0%BA%D0%B8-%D0%BF%D0%BE%D0%BF%D0%B0%D0%B4%D0%B0%D0%BB%D0%B8-%D0%B2-%D1%82%D0%B5%D1%81%D1%82%D0%BE%D0%B2%D1%8B%D0%B5-%D0%B2%D1%8B%D0%B1%D0%BE%D1%80%D0%BA%D0%B8

Тебе может это понравится...

Добавить комментарий