【发布时间】:2021-08-24 05:44:39
【问题描述】:
假设我正在使用 GridSearchCV 搜索超参数,并且我也在使用 Pipeline 因为我(想我)想要预处理我的数据:
param_grid = {
'svc__gamma': np.linspace(0.2, 1, 5)
}
pipeline = Pipeline(steps=[('scaler', StandardScaler()), ('svc', SVC())])
search = GridSearchCV(pipeline, param_grid, cv=10)
search.fit(train_x, train_y)
有没有办法测试我的假设,即包含 scaler 步骤实际上是有帮助的(不仅仅是删除它并重新运行它)?
即,有没有办法写:
param_grid = {
'svc__gamma': np.linspace(0.2, 1, 5),
'scaler': [On, Off]
}
或者我应该以不同的方式来解决这个问题?
【问题讨论】:
标签: python python-3.x machine-learning scikit-learn gridsearchcv