ray-project / tune-sklearn

A drop-in replacement for Scikit-Learn’s GridSearchCV / RandomizedSearchCV -- but with cutting edge hyperparameter tuning techniques.
https://docs.ray.io/en/master/tune/api_docs/sklearn.html
Apache License 2.0
465 stars 52 forks source link

Fix pipeline_auto_early_stop #168

Closed Yard1 closed 3 years ago

Yard1 commented 3 years ago

An oversight in _PipelineTrainable caused the xgboost/lightgbm/catboost methods to try and get boosters from the Pipeline object instead of the actual estimator, causing an exception. This PR fixes that and adds unit tests to check for this in the future.

richardliaw commented 3 years ago

Awesome!!