ray/python/requirements/ml/requirements_tune.txt
Antoni Baum 6ba4513777
[tune] Load experiment into searcher (#21506)
This PR adds a new method to the Searcher class, add_evaluated_trials. This method wraps around add_evaluated_point and allows the user to pass a Trial, list of Trials or ExperimentAnalysis to load into the searcher. Furthermore, this PR updates the HEBO version to the latest and removes outdated documentation, and adds add_evaluated_point methods to Dragonfly and SkOpt searchers.
2022-01-11 15:58:20 -08:00

41 lines
888 B
Text

-r requirements_dl.txt
autogluon.core==0.1.0
ax-platform==0.1.9; python_version < '3.7'
ax-platform==0.2.1; python_version >= '3.7'
bayesian-optimization==1.2.0
ConfigSpace==0.4.18
dragonfly-opt==0.1.6
flaml==0.6.7
freezegun==1.1.0
gluoncv==0.10.1.post0
gpy==1.10.0
autorom[accept-rom-license]
gym[atari]>=0.21.0; python_version >= '3.7'
gym[atari]==0.19.0; python_version < '3.7'
h5py==3.1.0
hpbandster==0.7.4
HEBO==0.3.2
hyperopt==0.2.5
jupyter==1.0.0
kubernetes==17.17.0
lightgbm==3.2.1
matplotlib==3.3.4; python_version < '3.7'
matplotlib==3.4.3; python_version >= '3.7'
mlflow==1.21.0
mxnet==1.8.0.post0
nevergrad==0.4.3.post7
optuna==2.9.1
pytest-remotedata==0.3.2
lightning-bolts==0.4.0
pytorch-lightning==1.4.9
shortuuid==1.0.1
scikit-learn==0.24.2
scikit-optimize==0.8.1
sigopt==7.5.0
smart_open==5.1.0
timm==0.4.5
transformers==4.10.0
wandb==0.12.5
xgboost==1.3.3
zoopt==0.4.1