如何绘制mean_test评分和mean_train评分GridSearchCV

Question

如何绘制支持向量机的mean_train_score和gamma值在GridSearchCV中的值？

Answer 1

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

例如，您可以使用C值作为行，使用gamma值作为列，并使用heatmap数组中每个元素的颜色强度与mean_test_score对应。

要实现这一点，首先需要创建如下所示的pandas.DataFrame：

\begin{array}{c | c c c} & C & gamma & mean\_test\_score \\ \hline 1 & 0.1 & 0.001 & 0.798 \\ 2 & 1 & 0.001 & 0.813 \\ 3 & 1 & 0.01 & 0.801 \\ 4 & 10 & 0.001 & 0.787 \\ \end{array}

要做到这一点，您需要将您所做的每一次运行都存储在一个不同的行中，其中将包含所有必要的超参数和结果。然后，您将需要创建一个pivot表，它将使用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