help default ReinforcementLearner users by assigning the model_type automatically

This commit is contained in:
robcaulk 2022-11-13 17:12:17 +01:00
parent 90f168d1ff
commit b421521be3

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')