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

[feature] Support any Searcher/Stopper in Tune #187

Closed Yard1 closed 3 years ago

Yard1 commented 3 years ago

The user should be able to pass a Searcher/Stopper object in addition to predefined strings in TuneSearchCV. It would be quite easy to add support for that, however first the search space conversions in tune-sklearn need to be unified as to either convert the objects that can be passed right now to Tune search spaces (which will then get converted to Searcher-specific spaces inside those) or make an API breaking change and allow only Tune search spaces + whatever can be passed to sklearn's GridSearchCV.

Yard1 commented 3 years ago

@richardliaw

richardliaw commented 3 years ago

Fixed with #198