如何绘制 GridSearchCV 的 mean_test 分数和 mean_train 分数

数据挖掘 支持向量机 网格搜索
2021-10-03 14:33:50

如何在SVM的值和值中绘制mean_train_scoremean_test_score值?GridSearchCVCgamma

1个回答

您可以将它们可视化为热图

例如,您可以将C值用作行,将gamma值用作列,并且热图数组中每个元素的颜色强度将对应于mean_test_score.

要实现这一点,您首先需要创建一个pandas.DataFrame这样的:

Cgammamean_test_score10.10.0010.798210.0010.813310.010.8014100.0010.787

为此,您需要将每次运行存储在不同的行中,其中将包含所有必要的超参数和结果。然后,您需要制作一个数据透视表,C用作行、gamma列和mean_test_score值。

pivot = pd.pivot_table(df, values=df['mean_test_score'])

这个枢轴将是形成热图的数组。现在您应该选择您的美学参数(例如颜色图)并继续制作热图。

sns.heatmap(pivot) # plus any other aesthetic parameters you wish