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 REQUEST] Allow for additional kw_args to tune.run #182

Closed Filco306 closed 3 years ago

Filco306 commented 3 years ago

Hello,

I am currently using the TuneSearchCV for some experiments that I have, and I am running into this issue, which I cannot override currently as there is no way to pass additional, customized arguments to tune.run() as it seems. Is there a way to allow this, or have I missed something in the documentation?

Thank you!

Cheers, Filip

Filco306 commented 3 years ago

I can close this now; for people running into the same problem, this can be used by defining the environment variable TUNE_GLOBAL_CHECKPOINT_S