我正在使用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]
}