From f6e8b5e882791ca0bfaf5ba299474cb9b6c2d47c Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Sat, 1 Mar 2025 15:22:21 +0100 Subject: [PATCH] refactor(reforcexy): refine hyperopt enablement condition 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 | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 6b68b12..d5a66ad 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -122,9 +122,11 @@ class ReforceXY(BaseReinforcementLearningModel): self.progressbar_callback: Optional[ProgressBarCallback] = None # Optuna hyperopt self.rl_config_optuna: dict = self.freqai_info.get("rl_config_optuna", {}) - self.hyperopt: bool = self.freqai_info.get( - "enabled", False - ) and self.rl_config_optuna.get("enabled", False) + self.hyperopt: bool = ( + self.freqai_info.get("enabled", False) + and self.rl_config_optuna.get("enabled", False) + and self.data_split_parameters.get("test_size", 0.1) > 0 + ) self.optuna_timeout_hours: float = self.rl_config_optuna.get("timeout_hours", 0) self.optuna_n_trials: int = self.rl_config_optuna.get("n_trials", 100) self.optuna_n_startup_trials: int = self.rl_config_optuna.get( -- 2.43.0