search

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

Gpu/cpu support #34

Closed inventormc closed 4 years ago

inventormc commented 4 years ago

Added support for user to specify number of gpu and cpu. This removed the need for n_jobs; need to confirm that these new changes behave correctly/don't negatively impact performance.

richardliaw commented 4 years ago

Hey, I think instead we should use num_jobs (keeping existing behavior), but also enable use_gpu. This makes it much easier to go from scikit learn to Tune.