9. Установка параметров
params = {
'objective': 'binary', # Задача двоичной классификации
'device': 'gpu', # Использовать обучение графического процессора
'metric': 'logloss', # Метрикой оценки является logloss
'boosting_type': 'gbdt',
'num_leaves': 31, # Установите количество листьев дерева
'learning_rate': 0,05, # Скорость обучения
'feature_fraction': 0,9 # Соотношение подмножества функций
10. Используйте метод train LightGBM для обучения модели и добавьте раннюю остановку
lgb_model = lgb.train(
params,
train_data,
num_boost_round=1000, #Максимальное количество итераций
valid_sets=[val_data], #Передавать только набор проверки
valid_names=['valid'], #Задать имя набора проверки
обратные вызовы =[lgb .early_stopping(stopping_rounds=50), # Ранняя остановка (нет улучшения в течение 50 раундов)
lgb.log_evaluation(10)], # Отображение журнала каждые 10 раундов
)
Мне интересно, использую ли я неправильную функцию или устаревшую функцию, и это вызывает ошибку.
Подробнее здесь: https://stackoverflow.com/questions/792 ... dataset-an
Как устранить эту ошибку «ValueError: для ранней остановки для оценки требуется хотя бы один набор данных и метрика оцен ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение