adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve the RL learning process

Browse Source 
Improve the RL learning process by selecting random start point for the agent, it can help to block the agent to only learn on the selected period of time, while improving the quality of the model.
This commit is contained in:
richardjozsa 2022-11-27 20:43:50 +01:00
parent 320535a227
commit 64d4a52a56
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/RL/BaseEnvironment.py
View File

@ -9,6 +9,7 @@ import pandas as pd
from gym import spaces from gym import spaces
from gym.utils import seeding from gym.utils import seeding
from pandas import DataFrame from pandas import DataFrame
import random
from freqtrade.data.dataprovider import DataProvider from freqtrade.data.dataprovider import DataProvider
@ -121,6 +122,9 @@ class BaseEnvironment(gym.Env):
self._done = False self._done = False
if self.starting_point is True: if self.starting_point is True:
length_of_data = int(self._end_tick/4)
start_tick = random.randint(self.window_size+1, length_of_data)
self._start_tick = start_tick
self._position_history = (self._start_tick * [None]) + [self._position] self._position_history = (self._start_tick * [None]) + [self._position]
else: else:
self._position_history = (self.window_size * [None]) + [self._position] self._position_history = (self.window_size * [None]) + [self._position]