如何用python画好confusion matrix

最新推荐文章于 2024-07-23 03:56:36 发布

leonordo

最新推荐文章于 2024-07-23 03:56:36 发布

阅读量4w

点赞数 11

分类专栏：机器学习文章标签： python confusion matrix matplotlib

本文链接：https://blog.csdn.net/guoyilin/article/details/42047615

版权

本文介绍如何利用matplotlib和scikit-learn在Python中绘制混淆矩阵。文章中提到，标准的confusion matrix图存在一些问题，如刻度位于方格中间、标签显示不理想、无法在方格内展示准确率等。作者提供了改进的代码，解决了这些问题，包括隐藏刻度、将标签置于方格中心、显示每个单元格的准确率，以及处理长标签的显示。完整代码可在作者的GitHub仓库中找到。

摘要由CSDN通过智能技术生成

在做分类的时候，经常需要画混淆矩阵，下面我们使用python的matplotlib包，scikit-learning机器学习库也同样提供了例子：http://scikit-learn.org/stable/auto_examples/plot_confusion_matrix.html，但是这样的图并不能满足我们的要求，

首先是刻度的显示是在方格的中间，这需要隐藏刻度，其次是如何把每个label显示在每个方块的中间，其次是如何在每个方格中显示accuracy数值，最后是如何在横坐标和纵坐标显示label的名字，在label name比较长的时候，如何处理显示问题。

直接贴上代码：下载见 https://github.com/guoyilin/plot.git

'''compute confusion matrix
labels.txt: contain label name.
predict.txt: predict_label true_label
'''
from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt
import numpy as np
#load labels.
labels = []
file = open('labels.tx