【发布时间】:2019-04-11 22:57:28
【问题描述】:
假设我有一个 sklearn 管道:
A(X) -> B(A(X)) -> C(B(A(X)))
grid_search = GridSearchCV(pipeline, param_grid)
我对超参数运行网格搜索以找到最佳模型。是否有内置功能来输出模型的中间和最终步骤?我想我可以使用在相同数据上发现的最佳参数以及每个步骤的结果重新运行管道,但也许这已经存在,例如:
grid_search.best_params.X_output_()
或
grid_search.step_A.output_()
我的最终目标是在每个转换步骤中可视化发现的最佳参数如何影响输入数据X。
【问题讨论】:
-
不,这不存在。管道有some functionality of caching 内部转换器,但由于 GridSearchCV 会将数据分成多个部分(根据
cv参数),我看不出它(或您想要的任何其他东西)在这里是否有用。 -
最佳参数是从这些参数在不同数据折叠中的平均性能中找到的。因此,试图将这些事情形象化并不容易。您最好对您想要查看的数据(可能是也可能不是相同的数据)重新运行管道
标签: python scikit-learn grid-search