Directly set model_type in base RL model

2022-11-28 16:02:17 +03:00
parent 1cdf5e0cfd
commit 9cbfa12011
2 changed files with 2 additions and 6 deletions
--- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
+++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
@@ -64,6 +64,7 @@ class BaseReinforcementLearningModel(IFreqaiModel):
        self.policy_type = self.freqai_info['rl_config']['policy_type']
        self.unset_outlier_removal()
        self.net_arch = self.rl_config.get('net_arch', [128, 128])
+        self.dd.model_type = "stable_baselines"

    def unset_outlier_removal(self):
        """