From 8e3440c130a84e01bf31dee24df61c90b1ec2809 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Mon, 21 Jul 2025 13:58:03 +0200 Subject: [PATCH] refactor(reforcexy): improve logging messages MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/user_data/freqaimodels/ReforceXY.py | 18 +++++------------- 1 file changed, 5 insertions(+), 13 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 18779cc..a13e08d 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -334,22 +334,14 @@ class ReforceXY(BaseReinforcementLearningModel): (train_timesteps * train_cycles + self.n_envs - 1) // self.n_envs ) * self.n_envs train_days = steps_to_days(train_timesteps, self.config.get("timeframe")) + test_days = steps_to_days(test_timesteps, self.config.get("timeframe")) total_days = steps_to_days(total_timesteps, self.config.get("timeframe")) logger.info("Action masking: %s", self.is_maskable) - logger.info( - "Train: %s steps (%s days) * %s cycles = Total %s (%s days)", - train_timesteps, - train_days, - train_cycles, - total_timesteps, - total_days, - ) - logger.info( - "Test: %s steps (%s days)", - test_timesteps, - steps_to_days(test_timesteps, self.config.get("timeframe")), - ) + logger.info("Train: %s steps (%s days)", train_timesteps, train_days) + logger.info("Train cycles: %s", train_cycles) + logger.info("Train total: %s steps (%s days)", total_timesteps, total_days) + logger.info("Test: %s steps (%s days)", test_timesteps, test_days) logger.info("Hyperopt: %s", self.hyperopt) start_time = time.time() -- 2.43.0