13-6 ada boosting和gradiesnt boosting
Last updated
Last updated
from sklearn.ensemble import AdaBoostClassifier
from sklearn.tree import DecisionTreeClassifier
# 使用决策树作为基础模型,决策树的参数在这里都能用
ada_clf = AdaBoostClassifier(DecisionTreeClassifier(max_depth=2), n_estimators=500)
ada_clf.fit(X_train, y_train)
ada_clf.score(X_test, y_test)from sklearn.ensemble import GradientBoostingClassifier
# 指定以决策树为基础,不能选择
gd_clf = GradientBoostingClassifier(max_depth=2, n_estimators=500)
gd_clf.fit(X_train, y_train)
gd_clf.score(X_test, y_test)from sklearn.ensemble import AdaBoostRegressor
from sklearn.ensemble import GradientBoostingRegressor