ENH Improve speed plot_adaboost_multiclass.py (#21651)

Sven Eschlbeck · glemaitre · commit 9b34ed57168f · 2021-12-25T12:26:49.000+01:00
diff --git a/examples/ensemble/plot_adaboost_multiclass.py b/examples/ensemble/plot_adaboost_multiclass.py
@@ -46,12 +46,12 @@
 y_train, y_test = y[:n_split], y[n_split:]
 
 bdt_real = AdaBoostClassifier(
-    DecisionTreeClassifier(max_depth=2), n_estimators=600, learning_rate=1
+    DecisionTreeClassifier(max_depth=2), n_estimators=300, learning_rate=1
 )
 
 bdt_discrete = AdaBoostClassifier(
     DecisionTreeClassifier(max_depth=2),
-    n_estimators=600,
+    n_estimators=300,
     learning_rate=1.5,
     algorithm="SAMME",
 )

Original file line number	Diff line number	Diff line change
`@@ -46,12 +46,12 @@`
`46`	`46`	`y_train, y_test = y[:n_split], y[n_split:]`
`47`	`47`
`48`	`48`	`bdt_real = AdaBoostClassifier(`
`49`		`- DecisionTreeClassifier(max_depth=2), n_estimators=600, learning_rate=1`
	`49`	`+ DecisionTreeClassifier(max_depth=2), n_estimators=300, learning_rate=1`
`50`	`50`	`)`
`51`	`51`
`52`	`52`	`bdt_discrete = AdaBoostClassifier(`
`53`	`53`	`DecisionTreeClassifier(max_depth=2),`
`54`		`- n_estimators=600,`
	`54`	`+ n_estimators=300,`
`55`	`55`	`learning_rate=1.5,`
`56`	`56`	`algorithm="SAMME",`
`57`	`57`	`)`