TuneBaseSearchCV to extend BaseSearchCV from sklearn

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

TuneBaseSearchCV to extend BaseSearchCV from sklearn #193

Closed Matgrb closed 3 years ago

Matgrb commented 3 years ago

Currently, the TuneBaseSearchCV inherits from BaseEstimator from sklearn, however, I suppose it should extend BaseSearchCV, in order to be fully consistent with the sklearn API. Thanks to this one can detect, whether this object is in fact an optimization search schema, or a classifier.

Yard1 commented 3 years ago

Will be covered by https://github.com/ray-project/tune-sklearn/pull/192