在sklearn中建立adaboost模型:
>>> from sklearn.ensemble import AdaBoostClassifier
>>> model = AdaBoostClassifier()
>>> model.fit(x_train, y_train)
>>> model.predict(x_test)
超参数
在实际操作中,最常见的超参数为:
- base_estimator: 弱学习器使用的模型(切勿忘记导入该模型)。
- n_estimators: 使用的弱学习器的最大数量。
举个例子:
from sklearn.tree import DecisionTreeClassifier
>>> model = AdaBoostClassifier(base_estimator = DecisionTreeClassifier(max_depth=2), n_estimators = 4)