]> Piment Noir Git Repositories - freqai-strategies.git/commitdiff
feat(weights): add _strip_label_weight_columns helper for find_labels collision avoidance
authorJérôme Benoit <jerome.benoit@piment-noir.org>
Sun, 24 May 2026 00:00:58 +0000 (02:00 +0200)
committerJérôme Benoit <jerome.benoit@piment-noir.org>
Sun, 24 May 2026 00:00:58 +0000 (02:00 +0200)
quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py

index f644f3f4bb16713d40e1596a1e6d3e51a8096b0a..661bce1b9fffcd7adaeb7c0f218df586565ee049 100644 (file)
@@ -102,6 +102,18 @@ class QuickAdapterRegressorV3(BaseRegressionModel):
 
     _TEST_SIZE: Final[float] = 0.1
 
+    _LABEL_WEIGHT_SUFFIX: Final[str] = "_weight"
+
+    @staticmethod
+    def _label_weight_column_name(label_col: str) -> str:
+        return f"{label_col}{QuickAdapterRegressorV3._LABEL_WEIGHT_SUFFIX}"
+
+    def _strip_label_weight_columns(self, dk: FreqaiDataKitchen) -> None:
+        dk.label_list = [
+            c for c in dk.label_list
+            if not c.endswith(self._LABEL_WEIGHT_SUFFIX)
+        ]
+
     _SQRT_2: Final[float] = np.sqrt(2.0)
 
     _OPTUNA_LABEL_N_OBJECTIVES: Final[int] = 7