From dc5766fb102af0c3b441e24278037ded3b841afb Mon Sep 17 00:00:00 2001 From: Shane Date: Fri, 24 May 2024 21:12:56 +1000 Subject: [PATCH] Update Base5ActionRLEnv.py Addition of action_space_type to support Discrete and Box action spaces. --- freqtrade/freqai/RL/Base5ActionRLEnv.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/freqtrade/freqai/RL/Base5ActionRLEnv.py b/freqtrade/freqai/RL/Base5ActionRLEnv.py index 2ae7f6b5a..13a376aa4 100644 --- a/freqtrade/freqai/RL/Base5ActionRLEnv.py +++ b/freqtrade/freqai/RL/Base5ActionRLEnv.py @@ -24,10 +24,16 @@ class Base5ActionRLEnv(BaseEnvironment): def __init__(self, **kwargs): super().__init__(**kwargs) + self.action_space_type = action_space_type self.actions = Actions def set_action_space(self): - self.action_space = spaces.Discrete(len(Actions)) + if self.action_space_type == "Discrete": + self.action_space = spaces.Discrete(len(Actions)) + elif self.action_space_type == "Box": + self.action_space = spaces.Box(low=-1, high=1, shape=(1,)) + else: + raise ValueError(f"Unknown action space type: {self.action_space_type}") def step(self, action: int): """