Skip to content Skip to sidebar Skip to footer

Lightgbm Early Stopping Not Working Propperly

I'm using light gbm for some machine learning task. I want to use early stopping to find the optimal number of trees given a number of hyperparameters. However, lgbm stops growing

Solution 1:

It is working properly : as said in doc for early stopping :

will stop training if one metric of one validation data doesn’t improve in last early_stopping_round rounds

and your logloss was better at round 1034.

Try to use first_metric_only = True or remove logloss from the list (using metric param)

Post a Comment for "Lightgbm Early Stopping Not Working Propperly"