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

Add `time_budget_s` parameter #134

Closed krfricke closed 3 years ago

krfricke commented 3 years ago

This passes the time_budget_s parameter to tune.run().

This also refactors parts of the base search _format_results() method to deal with missing values from trials (that e.g. have not been received due to early stopping/timeout).

krfricke commented 3 years ago

Added!