多分类混淆矩阵的理解

借用其它博客的一张例子示意图,该图为一个三分类问题的混淆矩阵,对角线的值表示分类器对该类别预测正确的个数,每一列纵轴表示这个类别真实的样本数,例如从第一列可以得知猫一共有10+3+5=18只,因此总样本数为三列之和,即66。

在这里插入图片描述
下面以猫为例,介绍从混淆矩阵中计算TP、TN、FP、FN。

TP(True Positive)FN(False Negative):从第一列可知,猫一共有18只,其中只有10个被分类器正确预测,而有3只被预测成狗,5只被预测成猪,因此对猫这个类别来说,TP=10(正确预测成猫的样本数)FN=8(被预测成其它类别的样本数)

FP(False Positive):从第一行可以得知分类器一共将10+1+2=13个样本预测成猫,其中正确的只有10个,错误的有3个,因此FP=3(被错误预测成猫的样本数)

TN(True Negative)TN表示分类器正确将不是猫的样本预测为其它类别的样本数,从第一行可知分类器将13个样本预测为猫,因此它将66-13=53个样本预测成了其它类别,而从第一列可知有8个样本本来是猫却被错误预测为其它类别(即FN),因此正确将不是猫的样本分成其它类别的有53-8=45个,整个计算公式就是 TN=66-13-8=66-(TP+FP+FN),所以TN一般会放在最后计算。

  • 16
    点赞
  • 115
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
混淆矩阵是用于评估分类模型性能的一种工具,它可以显示模型在不同类别上的预测结果和真实标签之间的对应关系。在PyTorch中,我们可以使用混淆矩阵来评估模型的分类准确性。 首先,我们需要导入必要的库和函数进行混淆矩阵的计算和可视化。可以参考和中的代码实现部分。 1. 数据集:在计算混淆矩阵之前,我们需要准备好一个验证集,该验证集包含模型预测的结果和真实标签。可以参考中的代码实现部分。 2. 代码:混淆矩阵类:在PyTorch中,可以通过编写一个混淆矩阵类来计算混淆矩阵。可以参考中的代码实现部分。 3. 在验证集上计算相关指标:使用混淆矩阵类计算验证集上的混淆矩阵,并计算相关指标,例如准确率、召回率、F1分数等。可以参考中的代码实现部分。 4. 结果:通过计算混淆矩阵和相关指标,我们可以得到模型在验证集上的分类性能结果。可以将混淆矩阵可视化,以更直观地理解模型的分类表现。可以参考中的代码实现部分。 绘制混淆矩阵的过程包括以下步骤: 1. 将混淆矩阵赋值给一个变量。 2. 打印混淆矩阵。 3. 使用imshow函数展示混淆矩阵,设置颜色变换从白色到蓝色。 4. 使用xticks函数将x轴的信息(0~num_classes-1)替换为标签的类别,并将x轴旋转45°。 5. 同理,使用yticks函数将y轴的信息替换为标签的类别。 6. 添加一个右侧颜色条,用来表示混淆矩阵中数值的密集程度,颜色越深表示数值越密集。 7. 设置横坐标为真实标签,纵坐标为预测标签。 8. 添加图像标题,例如"Confusion matrix"。 通过以上步骤,我们可以绘制出一个具有标签类别的混淆矩阵图像,该图像可以帮助我们更好地理解模型在不同类别上的分类表现。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* [混淆矩阵:用于多分类模型评估(pytorch)](https://blog.csdn.net/weixin_43760844/article/details/115208925)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *3* [人工智能学习07--pytorch11--分类网络:使用pytorch和tensorflow计算分类模型的混淆矩阵](https://blog.csdn.net/AMWICD/article/details/129443938)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值