]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
fix(qav3): uniformize optuna storage namespace
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Sat, 15 Feb 2025 22:47:39 +0000 (23:47 +0100)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Sat, 15 Feb 2025 22:47:39 +0000 (23:47 +0100)
Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py
quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py

index 04152a8caf6eb6fea92df037346f56dec849e616..11bdb94bcb16d6072afa99c130fdbedc3749a3ea 100644 (file)
@@ -230,11 +230,11 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel):
         storage_dir = str(dk.full_path)
         storage_backend = self.__optuna_config.get("storage", "file")
         if storage_backend == "sqlite":
-            storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair)}.sqlite"
+            storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair.split('/')[0])}.sqlite"
         elif storage_backend == "file":
             storage = optuna.storages.JournalStorage(
                 optuna.storages.journal.JournalFileBackend(
-                    f"{storage_dir}/optuna-{sanitize_path(dk.pair)}.log"
+                    f"{storage_dir}/optuna-{sanitize_path(dk.pair.split('/')[0])}.log"
                 )
             )
         return storage
index 0872136c9cdb2f21bf930decdfa2efae9778c377..fe44dd9acb262ba3c87757d5bd9cad2abdc743ea 100644 (file)
@@ -231,11 +231,11 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel):
         storage_dir = str(dk.full_path)
         storage_backend = self.__optuna_config.get("storage", "file")
         if storage_backend == "sqlite":
-            storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair)}.sqlite"
+            storage = f"sqlite:///{storage_dir}/optuna-{sanitize_path(dk.pair.split('/')[0])}.sqlite"
         elif storage_backend == "file":
             storage = optuna.storages.JournalStorage(
                 optuna.storages.journal.JournalFileBackend(
-                    f"{storage_dir}/optuna-{sanitize_path(dk.pair)}.log"
+                    f"{storage_dir}/optuna-{sanitize_path(dk.pair.split('/')[0])}.log"
                 )
             )
         return storage