Early stopping support for LightGBM and CatBoost

ray-project / tune-sklearn

A drop-in replacement for Scikit-Learn’s GridSearchCV / RandomizedSearchCV -- but with cutting edge hyperparameter tuning techniques.

Apache License 2.0

465 stars 52 forks source link

Closed Yard1 closed 4 years ago

Yard1 commented 4 years ago

Adds early stopping support to LightGBM and CatBoost using the same principle as the existing support for XGBoost.

Also replaced hardcoded Enum values with Enum.auto().

Unit tests and an example are included.

Yard1 commented 4 years ago

Only Mac failing, again.

richardliaw commented 4 years ago

looks good to me, @Yard1 ! great work - just one comment about inspect. @inventormc can you merge when it looks good?

Yard1 commented 4 years ago

@richardliaw @inventormc Inspect implemented!