From f7e87bb5d02e2a1be33ec489968b31bee905066f Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Sat, 17 May 2025 14:53:07 +0200 Subject: [PATCH] refactor(qav3): cleanup pivot labeling defaults handling MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/QuickAdapterRegressorV3.py | 4 ++-- quickadapter/user_data/strategies/Utils.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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: -- 2.43.0