DOC speed up plot_ensemble_oob.py (scikit-learn#21730)

siavrez · samronsin · commit 03f8485b8964 · 2021-11-30T17:28:12.000+01:00
* After 125, more estimators does not help the model and it improves  runtime by almost 50%

* accelerate plot_ensemble_oob.py changed max_estimator to 150 with step=5
diff --git a/examples/ensemble/plot_ensemble_oob.py b/examples/ensemble/plot_ensemble_oob.py
@@ -81,10 +81,10 @@
 
 # Range of `n_estimators` values to explore.
 min_estimators = 15
-max_estimators = 175
+max_estimators = 150
 
 for label, clf in ensemble_clfs:
-    for i in range(min_estimators, max_estimators + 1):
+    for i in range(min_estimators, max_estimators + 1, 5):
         clf.set_params(n_estimators=i)
         clf.fit(X, y)
 
