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

Does HEBOSearch work in tune-sklearn? #203

Closed qo4on closed 3 years ago

qo4on commented 3 years ago

Setting search_optimization=HEBOSearch() returns an error in your example.

Yard1 commented 3 years ago

There's only support for some searchers built into tune-sklearn - I've got a PR (https://github.com/ray-project/tune-sklearn/pull/198) to allow it to use any searcher in Ray but there are some design issues that need to be resolved.