Add constant features treatment

Doctorado-ML · Oct 13, 2023 · 0fdd754 · 0fdd754
1 parent 7035cc4
commit 0fdd754
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mufs/Selection.py b/mufs/Selection.py
@@ -172,6 +172,10 @@ def cfs(self, X, y):
                     id_selected = idx
                     merit = merit_new
                 candidates.pop()
+            if id_selected is None:
+                # No more features to add all merits are nan because of
+                # constant features
+                break
             candidates.append(feature_order[id_selected])
             self._scores.append(merit)
             del feature_order[id_selected]