我已经使用seaborn将我的数据绘制成一个混乱矩阵,但我遇到了一个问题。问题是它只在两个轴上显示从0到11的数字,因为我有12个不同的标签
我的代码如下所示:
cf_matrix = confusion_matrix(y_test, y_pred)
fig, ax = plt.subplots(figsize=(15,10))
sns.heatmap(cf_matrix, linewidths=1, annot=True, ax=ax, fmt='g')
在这里您可以看到我的混淆矩阵:
我得到了我应该得到的混淆矩阵。唯一的问题是没有显示的标签名称。我在互联网上搜索了很长一段时间,运气不好。是否有任何参数可以附加标签,或者如何实现
有人能帮我吗
谢谢大家!
//拉斯穆斯
当您对类别进行因子分解时,您应该保留级别,因此您可以将其与
pd.crosstab
而不是confusion_matrix
结合使用来绘制。以iris为例:在这一部分中,您将得到[0,…1,…2]中的标签y和级别,作为0,1,2对应的原始标签:
因此,我们非常适合您的要求:
和具有0,1,2的混淆矩阵:
我们返回并使用级别:
标签按字母顺序排列。 因此,使用numpy来区分“真”标签,您将得到一个按字母顺序排序的数组
相关问题 更多 >
编程相关推荐