From 70b25461f06b6a555d2cc21c3910834514df996a Mon Sep 17 00:00:00 2001 From: sonnhfit Date: Sun, 14 Aug 2022 20:47:58 +0700 Subject: [PATCH] add rl dependency --- requirements-freqai.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/requirements-freqai.txt b/requirements-freqai.txt index 26e4617af..8d8bb03c5 100644 --- a/requirements-freqai.txt +++ b/requirements-freqai.txt @@ -6,3 +6,6 @@ scikit-learn==1.1.2 joblib==1.1.0 catboost==1.0.6; platform_machine != 'aarch64' lightgbm==3.3.2 +torch==1.12.1 +stable-baselines3==1.5.0 +gym==0.21.0 \ No newline at end of file