[WIP] Triage bug with max_features

[no ci]
aramis-lab · Feb 23, 2024 · 2e101b8 · 2e101b8
1 parent c6ad37b
commit 2e101b8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/clinica/pipelines/machine_learning/algorithm.py b/clinica/pipelines/machine_learning/algorithm.py
@@ -364,6 +364,8 @@ def _launch_random_forest(
         min_samples_split,
         max_features,
     ):
+        assert max_features != "auto"
+
         if self._algorithm_params["balanced"]:
             classifier = RandomForestClassifier(
                 n_estimators=n_estimators,