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

help default ReinforcementLearner users by assigning the model_type automatically

Browse Source
This commit is contained in:
robcaulk 2022-11-13 17:12:17 +01:00
parent 90f168d1ff
commit b421521be3
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
freqtrade/freqai/data_drawer.py
View File

@ -99,10 +99,9 @@ class FreqaiDataDrawer:
"model_filename": "", "trained_timestamp": 0,
"data_path": "", "extras": {}}
self.limit_ram_use = self.freqai_info.get('limit_ram_usage', False)
if 'rl_config' in self.freqai_info:
self.model_type = 'stable_baselines'
logger.warning('User indicated rl_config, FreqAI will now use stable_baselines3'
' to save models.')
if 'Reinforcement' in self.config['freqaimodel']:
logger.warning('User passed a ReinforcementLearner model, FreqAI will '
'now use stable_baselines3 to save models.')
else:
self.model_type = self.freqai_info.get('model_save_type', 'joblib')