美文网首页
使用pytorch计算分类模型的混淆矩阵

使用pytorch计算分类模型的混淆矩阵

作者: 过气海豹 | 来源:发表于2021-09-20 23:37 被阅读0次

1.概念

混淆矩阵是评判模型结果的一种指标,属于模型评估的一部分,常用于评判分类器模型的优劣。
一个例子:


混淆矩阵例子

横坐标:每一列对应属于该类的所有验证样本(真实值)
纵坐标:每一行对应预测属于该类的所有样本(预测值)

对角线:预测正确的样本个数
预测值在对角线上分布的越密集(即对角线上的数越大),模型预测效果越好

2.准确率

准确率(Accuracy):预测正确的样本个数 / 所有样本总数(即矩阵中所有对角线数之和除以矩阵所有数字之和)

3.精确率、召回率(灵敏度)、特异度

tips:针对于每个类别,而准确率针对的是所有样本

3.1引入

二分类
表格元素解释

TP、TN、FP、FN总结:
TP、TN越大越好;
FN、FN越小越好

记法:第一个字母代表对错(True、False),第二个字母代表预测的值(Positive、Negative),如TP代表预测值为1,且预测正确。FN代表预测值为0,且预测错误。

3.2指标计算方法

指标

精确率(Precision):模型预测的所有positive中,预测正确的比例
召回率(Recall):所有真实的positive中,模型预测正确的positive比例
特异度(Specificity):所有真实negative中,模型预测正确的negative比例

参考博文:
1.https://blog.csdn.net/Orange_Spotty_Cat/article/details/80520839

参考视频:
1.https://www.bilibili.com/video/BV1GV411C7AW?spm_id_from=333.999.0.0

相关文章

  • 使用pytorch计算分类模型的混淆矩阵

    1.概念 混淆矩阵是评判模型结果的一种指标,属于模型评估的一部分,常用于评判分类器模型的优劣。一个例子: 横坐标:...

  • 混淆矩阵与精确率

    机器学习中,常使用混淆矩阵来对分类模型的效果进行衡量,本文主要讲述一下标准的二分类混淆矩阵与多分类混淆矩阵、准确率...

  • 2018-04-21FCN结果分析

    一、混淆矩阵confusion matrix 1,混淆矩阵 2,分类器评估方法:准确率和混淆矩阵*** 使用混淆矩...

  • 混淆矩阵

    简介 混淆矩阵是衡量分类模型准确度中最基本、直观,计算最简单的方法。混淆矩阵就是一张表,展示预测值各类归对、归错的...

  • 混淆矩阵confusion_matrix函数

    1.混淆矩阵 混淆矩阵是机器学习中总结分类模型预测结果的情形分析表,以矩阵形式将数据集中的记录按照真实的类别与分类...

  • 分类评价指标

    1. 混淆矩阵(Confusion matrix) 定义: 混淆矩阵是数据科学、数据分析和机器学习中分类模型预测结...

  • 10-分类模型的评估

    分类模型的评估 准确率 pass 混淆矩阵 在分类任务下,预测结果(Predicted Condition)与正确...

  • 分类模型的性能评估:混淆矩阵

    混淆矩阵(Confusion Matrix),也叫误差矩阵,是分类模型中常见的概念,在风控领域,我们常用「准确率」...

  • GCN图顶点分类理解

    GCN输出的H'矩阵,最后怎么令其作节点分类。即,GCN输出H’如何让节点分类的? 以pytorch的GCN模型为...

  • 模型性能评估

    不同的机器学习任务对应有不同的评估指标: 评估分类模型 混淆矩阵(Confusion matrix): True ...

网友评论

      本文标题:使用pytorch计算分类模型的混淆矩阵

      本文链接:https://www.haomeiwen.com/subject/anppgltx.html