adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add sb3 learn progress bar

Browse Source
This commit is contained in:
initrv
2023-04-02 02:59:02 +03:00
parent dc7e834911
commit cab82e8e60
5 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
freqtrade/constants.py
View File

@@ -599,6 +599,7 @@ CONF_SCHEMA = {
"policy_type": {"type": "string", "default": "MlpPolicy"},
"net_arch": {"type": "array", "default": [128, 128]},
"randomize_startinng_position": {"type": "boolean", "default": False},
"progress_bar": {"type": "boolean", "default": False},
"model_reward_parameters": {
"type": "object",
"properties": {

3
freqtrade/freqai/prediction_models/ReinforcementLearner.py
View File

@@ -71,7 +71,8 @@ class ReinforcementLearner(BaseReinforcementLearningModel):
model.learn(
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"]
)
if Path(dk.data_path / "best_model.zip").is_file():