make base 3ac and base 5ac environments. TDQN defaults to 3AC.

This commit is contained in:
robcaulk
2022-08-15 12:13:37 +02:00
parent 096533bcb9
commit 926023935f
5 changed files with 417 additions and 223 deletions

View File

@@ -8,7 +8,7 @@ from pandas import DataFrame
from abc import abstractmethod
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.freqai_interface import IFreqaiModel
from freqtrade.freqai.RL.BaseRLEnv import BaseRLEnv, Actions, Positions
from freqtrade.freqai.RL.Base3ActionRLEnv import Base3ActionRLEnv, Actions, Positions
from freqtrade.persistence import Trade
logger = logging.getLogger(__name__)
@@ -165,7 +165,7 @@ class BaseReinforcementLearningModel(IFreqaiModel):
hist_preds_df[return_str] = 0
class MyRLEnv(BaseRLEnv):
class MyRLEnv(Base3ActionRLEnv):
def step(self, action):
self._done = False