diff --git a/environment.yml b/environment.yml index 4f72dedc..99904480 100644 --- a/environment.yml +++ b/environment.yml @@ -11,12 +11,8 @@ dependencies: - requests - tqdm >= 4.37.0 - keras - - scikit-learn - psutil - - torch - - xgboost - beautifulsoup4 - - qhoptim - lxml - pytz - nodejs diff --git a/requirements.txt b/requirements.txt index 3e8057dc..b23c3151 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,9 @@ tensorboard >= 2.3 tensorflow >= 2.3 +qhoptim >= 1.1 +torch >= 1.6 +scikit-learn +xgboost atoma scikit-optimize envlogger