From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Sat, 17 May 2025 12:53:07 +0000 (+0200)
Subject: refactor(qav3): cleanup pivot labeling defaults handling
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=f7e87bb5d02e2a1be33ec489968b31bee905066f;p=freqai-strategies.git

refactor(qav3): cleanup pivot labeling defaults handling

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---

diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
index a808420..80b44a8 100644
--- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
+++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
@@ -884,8 +884,8 @@ def zigzag(
 
     def calculate_depth(
         pivots_indices: list[int],
-        min_depth: int = 6,
-        max_depth: int = 24,
+        min_depth: int = int(initial_depth / 2),
+        max_depth: int = initial_depth * 2,
         depth_scaling_factor: float = 0.75,
     ) -> int:
         if len(pivots_indices) < 2:
diff --git a/quickadapter/user_data/strategies/Utils.py b/quickadapter/user_data/strategies/Utils.py
index 3c106fb..78e1269 100644
--- a/quickadapter/user_data/strategies/Utils.py
+++ b/quickadapter/user_data/strategies/Utils.py
@@ -380,8 +380,8 @@ def zigzag(
 
     def calculate_depth(
         pivots_indices: list[int],
-        min_depth: int = 6,
-        max_depth: int = 24,
+        min_depth: int = int(initial_depth / 2),
+        max_depth: int = initial_depth * 2,
         depth_scaling_factor: float = 0.75,
     ) -> int:
         if len(pivots_indices) < 2: