如何在SVM的值和值中绘制mean_train_score
和mean_test_score
值?GridSearchCV
C
gamma
如何绘制 GridSearchCV 的 mean_test 分数和 mean_train 分数
数据挖掘
支持向量机
网格搜索
2021-10-03 14:33:50
1个回答
您可以将它们可视化为热图。
例如,您可以将C
值用作行,将gamma
值用作列,并且热图数组中每个元素的颜色强度将对应于mean_test_score
.
要实现这一点,您首先需要创建一个pandas.DataFrame
这样的:
为此,您需要将每次运行存储在不同的行中,其中将包含所有必要的超参数和结果。然后,您需要制作一个数据透视表,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
其它你可能感兴趣的问题