From bffed94c192658dbdde27c10e70735fffb0d815c Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 12 Mar 2025 12:09:09 +0100 Subject: [PATCH] fix(qav3): ensure previous optuna study is loaded when continuous = False MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py | 2 ++ .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index 1081505..4b83eb6 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -318,6 +318,7 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): pruner=pruner, direction=optuna.study.StudyDirection.MINIMIZE, storage=storage, + load_if_exists=not self.__optuna_config.get("continuous", True), ) if self.__optuna_config.get("warm_start", True): self.optuna_hp_enqueue_previous_best_trial(pair, study) @@ -396,6 +397,7 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): pruner=pruner, direction=optuna.study.StudyDirection.MINIMIZE, storage=storage, + load_if_exists=not self.__optuna_config.get("continuous", True), ) if self.__optuna_config.get("warm_start", True): self.optuna_period_enqueue_previous_best_trial(pair, study) diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index e7a46f5..7963245 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -321,6 +321,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): pruner=pruner, direction=optuna.study.StudyDirection.MINIMIZE, storage=storage, + load_if_exists=not self.__optuna_config.get("continuous", True), ) if self.__optuna_config.get("warm_start", True): self.optuna_hp_enqueue_previous_best_trial(pair, study) @@ -399,6 +400,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): pruner=pruner, direction=optuna.study.StudyDirection.MINIMIZE, storage=storage, + load_if_exists=not self.__optuna_config.get("continuous", True), ) if self.__optuna_config.get("warm_start", True): self.optuna_period_enqueue_previous_best_trial(pair, study) -- 2.43.0