From 7f991585313db21b7a36cc1d5a7b7b0abc55b25c Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 18 Feb 2025 16:50:35 +0100 Subject: [PATCH] fix(reforcexy): proper compute fit time 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 | 3 +++ 1 file changed, 3 insertions(+) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index c0b9aee..bd68492 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -331,6 +331,7 @@ class ReforceXY(BaseReinforcementLearningModel): ) logger.info("Hyperopt: %s", self.hyperopt) + start = time.time() if self.hyperopt: model_params = self.study(train_df, total_timesteps, dk) else: @@ -362,6 +363,8 @@ class ReforceXY(BaseReinforcementLearningModel): finally: if self.progressbar_callback: self.progressbar_callback.on_training_end() + time_spent = time.time() - start + self.dd.update_metric_tracker("fit_time", time_spent, dk.pair) if Path(dk.data_path / "best_model.zip").is_file(): logger.info("Callback found a best model.") -- 2.43.0