根据上面的@James Dellinger评论,并从那里扩展,我能够完成它。事实证明,“秘密酱汁”确实是一个几乎没有文档记录的特性-
__
(双下划线)分隔符(在
Pipeline
文档):似乎要添加内部/基本估计器名称,然后是
__
param_grid
其中包括外部和内部估计器的参数。
BaggingClassifier
RandomForestClassifier
. 因此,您需要做的是,首先,导入需要导入的内容:
from sklearn.ensemble import BaggingClassifier, RandomForestClassifier
from sklearn.model_selection import GridSearchCV
其次是
参数网格
param_grid = {
'bootstrap': [True, False],
'bootstrap_features': [True, False],
'n_estimators': [5, 10, 15],
'max_samples' : [0.6, 0.8, 1.0],
'base_estimator__bootstrap': [True, False],
'base_estimator__n_estimators': [100, 200, 300],
'base_estimator__max_features' : [0.6, 0.8, 1.0]
}
最后,您的网格搜索:
grid_search=GridSearchCV(BaggingClassifier(base_estimator=RandomForestClassifier()), param_grid=param_grid, cv=5)