lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 02:12:01 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update Base5ActionRLEnv.py

Browse Source 
Addition of action_space_type to support Discrete and Box action spaces.
This commit is contained in:
Shane 2024-05-24 21:12:56 +10:00 committed by GitHub
parent 07fba3abb0
commit dc5766fb10
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
freqtrade/freqai/RL/Base5ActionRLEnv.py
View File

@ -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):
"""