如何绘制 GridSearch 结果?

数据挖掘 支持向量机 交叉验证 matplotlib 网格搜索
2022-03-12 05:36:50

我用 GridSearch 训练了一个 SVM 模型

svc = SVC()
parameters = {
  'kernel': ['linear', 'rbf'],
  'C': [0.1, 1, 10]
}

cv = GridSearchCV(svc, parameters, cv=5)
cv.fit(v_train, y_train)

print_results(cv)

这是我得到的结果:

BEST PARAMS: {'C': 1, 'kernel': 'linear'}

0.912 (+/-0.037) for {'C': 0.1, 'kernel': 'linear'}
0.763 (+/-0.027) for {'C': 0.1, 'kernel': 'rbf'}
0.942 (+/-0.045) for {'C': 1, 'kernel': 'linear'}
0.903 (+/-0.044) for {'C': 1, 'kernel': 'rbf'}
0.94 (+/-0.043) for {'C': 10, 'kernel': 'linear'}
0.928 (+/-0.046) for {'C': 10, 'kernel': 'rbf'}

绘制此结果的最佳方法是什么?在一个包含C参数和kernel相应精度的图中。

谢谢你。

2个回答

我的建议是将结果作为数据框

pd.DataFrame(cv.cv_results_)

然后您将数据放在更易于处理的数据框中。

对于另一个问题(在您的评论中),一旦您将数据放入一个不错的数据框中,这就是数据可视化。你怎么能把它放在一个漂亮的可视化中?

更容易的是 2d 图或 3d 甚至等高线图。通过谷歌搜索它们和python,您将获得有关如何做的很好的教程

您可以使用简单的lineplot

在此处输入图像描述