Update mlxtend/feature_selection/exhaustive_feature_selector.py

Co-authored-by: Peter Gedeck <[email protected]>
rasbt · Nov 11, 2024 · d21c0f5 · d21c0f5
1 parent 51c4114
commit d21c0f5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mlxtend/feature_selection/exhaustive_feature_selector.py b/mlxtend/feature_selection/exhaustive_feature_selector.py
@@ -483,7 +483,11 @@ def ncr(n, r):
         return self
 
     def finalize_fit(self):
-        max_score = -np.inf
+        if np.__version__ < "2.0":
+            ninf = np.NINF
+        else:
+            ninf = -np.inf
+        max_score = ninf
         for c in self.subsets_:
             if self.subsets_[c]["avg_score"] > max_score:
                 best_subset = c