From db622b9ed84db114bbb617260ce817c55d27ad8e Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Mon, 10 Feb 2025 13:45:26 +0100 Subject: [PATCH] fix(qav3): ensure different study name per crypto pair 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 | 3 ++- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index 5b4a9fb..027412e 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -67,7 +67,8 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): start = time.time() if self.__optuna_hyperopt: - storage_dir, study_name = str(dk.full_path).rsplit("/", 1) + study_name = dk.pair + storage_dir = str(dk.full_path).rsplit("/", 1) pruner = optuna.pruners.HyperbandPruner() study = optuna.create_study( study_name=study_name, diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index 0f00dff..9bc07eb 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -67,7 +67,8 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): start = time.time() if self.__optuna_hyperopt: - storage_dir, study_name = str(dk.full_path).rsplit("/", 1) + study_name = dk.pair + storage_dir = str(dk.full_path).rsplit("/", 1) pruner = optuna.pruners.HyperbandPruner() study = optuna.create_study( study_name=study_name, -- 2.43.0