0%

Sklearn Pipeline GridSearchCV 使用

1. 介绍

Sklearn 的 GridSearchCV 和 Pipeline 能够合并多个模型,并且可以使用 FeatureUnion 合并自己 extraction 的 feature 的确是比较方便的。

本文主要在 MNIST 数据集上进行测试。使用比较简单的 KNN 算法进行预测。

2. 方法

主要代码在 Github 上。

使用 GridSearchCV 进行参数选择以及crossvalidation。注意 param_grid 也就是每个模型的参数选择。

Pipeline 可以把设置一个训练过程,如果有多个feature需要使用FeatureUnion 进行操作。


因为我们是朋友,所以你可以使用我的文字,但请注明出处:http://alwa.info