From 44e1a076829838a36be6e53b83326fc840f93b24 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Thu, 27 Mar 2025 10:20:20 +0100 Subject: [PATCH] perf(qav3): tune optuna default timeout MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- quickadapter/user_data/config-template.json | 2 +- .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py | 4 ++-- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/quickadapter/user_data/config-template.json b/quickadapter/user_data/config-template.json index 80e5eae..4d428c1 100644 --- a/quickadapter/user_data/config-template.json +++ b/quickadapter/user_data/config-template.json @@ -123,7 +123,7 @@ "enabled": true, "n_jobs": 6, "n_trials": 36, - "timeout": 3600, + "timeout": 7200, "candles_step": 10, "storage": "file" }, diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py index 107521c..e070591 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV3.py @@ -339,7 +339,7 @@ class LightGBMRegressorQuickAdapterV3(BaseRegressionModel): self.__optuna_config.get("n_jobs", 1), max(int(self.max_system_threads / 4), 1), ), - timeout=self.__optuna_config.get("timeout", 3600), + timeout=self.__optuna_config.get("timeout", 7200), gc_after_trial=True, ) except Exception as e: @@ -427,7 +427,7 @@ class LightGBMRegressorQuickAdapterV3(BaseRegressionModel): self.__optuna_config.get("n_jobs", 1), max(int(self.max_system_threads / 4), 1), ), - timeout=self.__optuna_config.get("timeout", 3600), + timeout=self.__optuna_config.get("timeout", 7200), gc_after_trial=True, ) except Exception as e: diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py index 4116bb0..f127eb9 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py @@ -342,7 +342,7 @@ class XGBoostRegressorQuickAdapterV3(BaseRegressionModel): self.__optuna_config.get("n_jobs", 1), max(int(self.max_system_threads / 4), 1), ), - timeout=self.__optuna_config.get("timeout", 3600), + timeout=self.__optuna_config.get("timeout", 7200), gc_after_trial=True, ) except Exception as e: @@ -430,7 +430,7 @@ class XGBoostRegressorQuickAdapterV3(BaseRegressionModel): self.__optuna_config.get("n_jobs", 1), max(int(self.max_system_threads / 4), 1), ), - timeout=self.__optuna_config.get("timeout", 3600), + timeout=self.__optuna_config.get("timeout", 7200), gc_after_trial=True, ) except Exception as e: -- 2.43.0