use rl_config get instead of freqai_info
This commit is contained in:
parent
4c608e7167
commit
cfc0410388
@ -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():
|
||||||
|
Loading…
Reference in New Issue
Block a user