ENH Improve speed plot_adaboost_multiclass.py
@@ -46,12 +46,12 @@ X_train, X_test = X[:n_split], X[n_split:]
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",
)
