From 13cd18dc9a84e36fbc11ed2d17a2a65ebdaecdb5 Mon Sep 17 00:00:00 2001 From: MukavaValkku Date: Mon, 15 Aug 2022 14:05:01 +0300 Subject: [PATCH] PPO policy change + verbose=1 --- .../freqai/prediction_models/ReinforcementLearningPPO.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/prediction_models/ReinforcementLearningPPO.py b/freqtrade/freqai/prediction_models/ReinforcementLearningPPO.py index cc56852df..5bc33bff1 100644 --- a/freqtrade/freqai/prediction_models/ReinforcementLearningPPO.py +++ b/freqtrade/freqai/prediction_models/ReinforcementLearningPPO.py @@ -53,8 +53,8 @@ class ReinforcementLearningPPO(BaseReinforcementLearningModel): policy_kwargs = dict(activation_fn=th.nn.ReLU, net_arch=[256, 256, 128]) - model = PPO('MultiInputPolicy', train_env, policy_kwargs=policy_kwargs, - tensorboard_log=f"{path}/ppo/tensorboard/", learning_rate=0.00025, gamma=0.9 + model = PPO('MlpPolicy', train_env, policy_kwargs=policy_kwargs, + tensorboard_log=f"{path}/ppo/tensorboard/", learning_rate=0.00025, gamma=0.9, verbose=1 ) model.learn(