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

ensure model_type is defined

Browse Source
This commit is contained in:
robcaulk 2022-11-13 18:50:25 +01:00
parent 6394ef4558
commit bf4d5b432a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
freqtrade/freqai/data_drawer.py
View File

@ -99,6 +99,7 @@ class FreqaiDataDrawer:
"model_filename": "", "trained_timestamp": 0, "model_filename": "", "trained_timestamp": 0,
"data_path": "", "extras": {}} "data_path": "", "extras": {}}
if 'Reinforcement' in self.config['freqaimodel']: if 'Reinforcement' in self.config['freqaimodel']:
self.model_type = 'stable_baselines'
logger.warning('User passed a ReinforcementLearner model, FreqAI will ' logger.warning('User passed a ReinforcementLearner model, FreqAI will '
'now use stable_baselines3 to save models.') 'now use stable_baselines3 to save models.')
else: else: