From a3b633f212bd947e5792404255c4d4c1e7ceedb8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 12 Feb 2025 23:23:38 +0100 Subject: [PATCH] refactor(qav3): rename storage optuna tunable 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 +- .../freqaimodels/LightGBMRegressorQuickAdapterV35.py | 6 +++--- .../freqaimodels/XGBoostRegressorQuickAdapterV35.py | 6 +++--- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/quickadapter/user_data/config-template.json b/quickadapter/user_data/config-template.json index ba7ba0e..c1d4c4f 100644 --- a/quickadapter/user_data/config-template.json +++ b/quickadapter/user_data/config-template.json @@ -122,7 +122,7 @@ "n_jobs": 6, "timeout": 3600, "candles_step": 10, - "storage_type": "sqlite" + "storage": "sqlite" }, "extra_returns_per_train": { "DI_value_param1": 0, diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index 86318b6..cc0ee08 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -234,10 +234,10 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): def get_optuna_storage(self, dk: FreqaiDataKitchen): storage_dir = str(dk.full_path) - storage_type = self.__optuna_config.get("storage_type", "sqlite") - if storage_type == "sqlite": + storage_backend = self.__optuna_config.get("storage", "sqlite") + if storage_backend == "sqlite": storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair)}.sqlite" - elif storage_type == "file": + elif storage_backend == "file": storage = optuna.storages.JournalStorage( optuna.storages.journal.JournalFileBackend( f"{storage_dir}/optuna-{sanitize_path(dk.pair)}.log" diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index 36363c8..63d89b8 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -237,10 +237,10 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): def get_optuna_storage(self, dk: FreqaiDataKitchen): storage_dir = str(dk.full_path) - storage_type = self.__optuna_config.get("storage_type", "sqlite") - if storage_type == "sqlite": + storage_backend = self.__optuna_config.get("storage", "sqlite") + if storage_backend == "sqlite": storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair)}.sqlite" - elif storage_type == "file": + elif storage_backend == "file": storage = optuna.storages.JournalStorage( optuna.storages.journal.JournalFileBackend( f"{storage_dir}/optuna-{sanitize_path(dk.pair)}.log" -- 2.43.0