numpy<=1.19.5
scikit-learn
pandas
matplotlib
scikit-optimize
joblib
requests
plotly
easy_mpl>=0.20.3

[all]
tensorflow
scikit-optimize
h5py<2.11.0
xgboost
lightgbm
catboost
tpot
imageio
pyshp
optuna
hyperopt
netCDF4
xarray
wrapt
seaborn

[eda]
seaborn
scikit-learn
easy_mpl>=0.20.3

[exp]
catboost
lightgbm
xgboost
tpot
optuna
hyperopt
scikit-optimize
h5py<2.11.0

[hpo]
optuna
hyperopt
scikit-optimize

[ml]
numpy
matplotlib
pandas
scikit-learn
xgboost
catboostlightgbm
easy_mpl>=0.20.3

[ml_hpo]
numpy
matplotlib
pandas
scikit-learn
xgboost
catboostlightgbm
easy_mpl>=0.20.3
hyperopt
scikit-optimize
optuna

[post_process]
lime
shap

[tf]
h5py<2.11.0
numpy<=1.19.5
easy_mpl
tensorflow
pandas
matplotlib
scikit-learn

[tf_hpo]
h5py<2.11.0
numpy<=1.19.5
easy_mpl
tensorflow
pandas
matplotlib
scikit-learn
hyperopt
scikit-optimize
optuna

[torch]
h5py
numpy
easy_mpl>=0.20.3
pytorch
pandas
matplotlib
scikit-learn

[torch_hpo_requires]
h5py
numpy
easy_mpl>=0.20.3
pytorch
pandas
matplotlib
scikit-learn
hyperopt
scikit-optimize
optuna
