DOC accelerate plot_successive_halving_iterations.py example #21598 (#21612)

sply88 · glemaitre · commit 5cca8bfbeb08 · 2021-12-25T12:26:49.000+01:00
* accelerate plot_successive_halving_iterations.py example #21598 * n_estimators back to 20
diff --git a/examples/model_selection/plot_successive_halving_iterations.py b/examples/model_selection/plot_successive_halving_iterations.py
@@ -27,13 +27,13 @@
 
 rng = np.random.RandomState(0)
 
-X, y = datasets.make_classification(n_samples=700, random_state=rng)
+X, y = datasets.make_classification(n_samples=400, n_features=12, random_state=rng)
 
 clf = RandomForestClassifier(n_estimators=20, random_state=rng)
 
 param_dist = {
     "max_depth": [3, None],
-    "max_features": randint(1, 11),
+    "max_features": randint(1, 6),
     "min_samples_split": randint(2, 11),
     "bootstrap": [True, False],
     "criterion": ["gini", "entropy"],
