Python绘制混淆矩阵

一、混淆矩阵介绍

二、绘制混淆矩阵

三、在深度学习代码中添加绘制混淆矩阵模块

一、混淆矩阵介绍

  混淆矩阵的每一列代表了预测类别，每一列的总数表示预测为该类别的数据的数目；每一行代表了数据的真实归属类别，每一行的数据总数表示该类别的数据实例的数目。每一列中的数值表示真实数据被预测为该类的数目。

  以下图为例，第一行的数值总和为2+0+0=2，表示ant类别共有2个样本，其中，有2个样本被预测为ant类别，0个样本被预测为bird类别，0个样本被预测为cat类别，即ant类别的图像全预测正确了。其他行同理。   上面这个混淆矩阵并没有归一化，对其进行归一化后的结果如下。以第三行为例进行解释：0.33表示有33%的cat图像被预测为了ant，0%的cat图像被预测为bird，也即没有cat图像被预测为bird，67%的cat图像被预测为cat。   上面说这么多，主要是想让大家直观地理解混淆矩阵到底是怎么一回事。总是，

查看原文