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

[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
SeqMetrics>=1.3.2

[exp]
catboost
lightgbm
xgboost
tpot
optuna
hyperopt
scikit-optimize
h5py<2.11.0
SeqMetrics>=1.3.2

[hpo]
optuna
hyperopt
scikit-optimize
SeqMetrics>=1.3.2

[ml]
numpy
matplotlib
pandas
scikit-learn
xgboost
catboost
lightgbm
easy_mpl>=0.20.3
SeqMetrics>=1.3.2

[ml_hpo]
numpy
matplotlib
pandas
scikit-learn
xgboost
catboost
lightgbm
easy_mpl>=0.20.3
hyperopt
scikit-optimize
optuna
SeqMetrics>=1.3.2

[post_process]
lime
shap
SeqMetrics>=1.3.2

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

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

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

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