Merge pull request #129 from EwoutH/patch-9

feature_scoring.py: Change `max_features` in get_rf_feature_scores to `"sqrt"`
quaquel · May 18, 2022 · 5e03fd9 · 5e03fd9
2 parents c1da7cf + 6cb9131
commit 5e03fd9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ema_workbench/analysis/feature_scoring.py b/ema_workbench/analysis/feature_scoring.py
@@ -161,7 +161,7 @@ def get_rf_feature_scores(
     y,
     mode=RuleInductionType.CLASSIFICATION,
     nr_trees=250,
-    max_features="auto",
+    max_features="sqrt",
     max_depth=None,
     min_samples_split=2,
     min_samples_leaf=1,