FIX SelectKBest

MannLabs · Apr 3, 2023 · db49723 · db49723
1 parent 0debb0f
commit db49723
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/omiclearn/utils/ml_helper.py b/omiclearn/utils/ml_helper.py
@@ -123,11 +123,11 @@ def select_features(feature_method, X, y, max_features, n_trees, random_state):
 
     elif "k-best" in feature_method:
         if feature_method == "k-best (mutual_info_classif)":
-            clf = SelectKBest(mutual_info_classif, max_features)
+            clf = SelectKBest(mutual_info_classif, k=max_features)
         elif feature_method == "k-best (f_classif)":
-            clf = SelectKBest(f_classif, max_features)
+            clf = SelectKBest(f_classif, k=max_features)
         elif feature_method == "k-best (chi2)":
-            clf = SelectKBest(chi2, max_features)
+            clf = SelectKBest(chi2, k=max_features)
         else:
             raise NotImplementedError(
                 "Feature method {} not implemented.".format(feature_method)