ray
ray copied to clipboard
[tune] remove dependency on gpy
Description
Replace GPy with scikit-learn's GaussianProcessRegressor in the PB2 (Population Based Bandits) scheduler.
Additional information
- Rewrote
TV_SquaredExpkernel to implement sklearn'sKernelinterface instead of GPy'sKern - Replaced
GPy.models.GPRegressionwithsklearn.gaussian_process.GaussianProcessRegressor - Removed
gpyfromtune-test-requirements.txt - Updated documentation to only require scikit-learn (which is already a Ray dependency)
Testing
Re-ran existing PB2 tests in est_trial_scheduler_pbt.py to validate change.