如何使用AdaBoost和GridSearchCV获得最佳估计器参数

时间:2019-05-06 01:08:23

标签: python adaboost gridsearchcv

我正在使用AdaBoost,并且我想看看使用GridSearchCV的估算器参数效果最好。是否可以在我的“ parameters”变量中包含估算器参数?例如,如何更改DecisionTreeClassifier的max_depth?

pipe = Pipeline([
             ('normalize', MinMaxScaler(feature_range=(0, 1))),
             ('scale', StandardScaler()),
             ('booster', AdaBoostClassifier()),
             ])

parameters = {
        "booster__base_estimator": [DecisionTreeClassifier(), RandomForestClassifier(n_estimators=20)],
        "booster__n_estimators": [20, 30, 40],
        "booster__learning_rate": [0.1, 0.5, 1]
     }

0 个答案:

没有答案