From: Jérôme Benoit Date: Thu, 27 Mar 2025 09:20:20 +0000 (+0100) Subject: perf(qav3): tune optuna default timeout X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=44e1a076829838a36be6e53b83326fc840f93b24;p=freqai-strategies.git perf(qav3): tune optuna default timeout Signed-off-by: Jérôme Benoit --- 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: