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

Pass estimators via object store IDs #196

Closed krfricke closed 3 years ago

krfricke commented 3 years ago

Currently estimator objects are passed via the config parameter (as is the data). However, in some cases this can lead to unexpected errors when creating the search space with Hyperopt.

By passing the estimators via the object store we circumvent this problem.

After the next release we might want to use tune.with_parameters() to avoid passing anything other than the search space via the config parameter, see https://github.com/ray-project/ray/pull/14532

Closes https://github.com/ray-project/tune-sklearn/issues/186