From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Thu, 19 Jun 2025 09:25:38 +0000 (+0200)
Subject: fix(qav3): kmeans MO selection clause guard on the number of samples
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=1ebb952cb75a739965c911c89f5546e5dde2cc95;p=freqai-strategies.git

fix(qav3): kmeans MO selection clause guard on the number of samples

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 157d616..c615b8c 100644
--- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
+++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
@@ -666,7 +666,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel):
                 return np.sum(np_weights * (ideal_point - normalized_matrix), axis=1)
             elif metric in {"kmeans", "kmeans2"}:
                 if n_samples < 2:
-                    return np.ndarray([])
+                    return np.full(n_samples, np.inf)
                 n_clusters = min(max(2, int(np.sqrt(n_samples / 2))), 10, n_samples)
                 if metric == "kmeans":
                     kmeans = sklearn.cluster.KMeans(