use rl_config get instead of freqai_info

This commit is contained in:
initrv 2023-04-02 04:08:07 +03:00
parent 4c608e7167
commit cfc0410388

View File

@ -72,7 +72,7 @@ class ReinforcementLearner(BaseReinforcementLearningModel):
model.learn( model.learn(
total_timesteps=int(total_timesteps), total_timesteps=int(total_timesteps),
callback=[self.eval_callback, self.tensorboard_callback], callback=[self.eval_callback, self.tensorboard_callback],
progress_bar=self.freqai_info["rl_config"]["progress_bar"] progress_bar=self.rl_config.get('progress_bar', False)
) )
if Path(dk.data_path / "best_model.zip").is_file(): if Path(dk.data_path / "best_model.zip").is_file():