Fix bug with max_points in MH when the number of maxima is small.

GAA-UAM · Jun 29, 2023 · e56b1b4 · e56b1b4
1 parent c4b2a66
commit e56b1b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/skfda/preprocessing/dim_reduction/variable_selection/maxima_hunting.py b/skfda/preprocessing/dim_reduction/variable_selection/maxima_hunting.py
@@ -101,7 +101,7 @@ def __call__(self, X: FDataGrid) -> NDArrayInt:
             values = X.data_matrix[:, indexes]
             partition_indexes = np.argpartition(
                 values,
-                -self.max_points,
+                -min(self.max_points, len(values)),
                 axis=None,
             )
             indexes = indexes[np.sort(partition_indexes[-self.max_points:])]