From b421521be34c1bbef5e9203eebfcd9ba20aaef28 Mon Sep 17 00:00:00 2001 From: robcaulk Date: Sun, 13 Nov 2022 17:12:17 +0100 Subject: [PATCH] help default ReinforcementLearner users by assigning the model_type automatically --- freqtrade/freqai/data_drawer.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py index d6ad3047d..d41675f89 100644 --- a/freqtrade/freqai/data_drawer.py +++ b/freqtrade/freqai/data_drawer.py @@ -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')