DOC speed up plot_ensemble_oob.py (#21730)

* 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
scikit-learn · Nov 24, 2021 · c9e5067 · c9e5067
1 parent 6a693ac
commit c9e5067
Showing 1 changed file with 2 additions and 2 deletions.
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)