From c83dd2d806ceda37c80485940c5e1ea62ffb69ae Mon Sep 17 00:00:00 2001 From: Shane Date: Fri, 24 May 2024 21:29:38 +1000 Subject: [PATCH] Update BaseReinforcementLearningModel.py Add support for DDPG and TD3. --- freqtrade/freqai/RL/BaseReinforcementLearningModel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py index 225ed3d50..61cef1ab0 100644 --- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py +++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py @@ -32,7 +32,7 @@ logger = logging.getLogger(__name__) torch.multiprocessing.set_sharing_strategy("file_system") -SB3_MODELS = ["PPO", "A2C", "DQN"] +SB3_MODELS = ["PPO", "A2C", "DQN", "DDPG", "TD3"] SB3_CONTRIB_MODELS = ["TRPO", "ARS", "RecurrentPPO", "MaskablePPO", "QRDQN"]