原文:对于计算正确率时 logits.argmax(dim=1),torch.eq(pre_label,label)

额 好像是一句非常简单的代码 ,但是作为新手 ,我是完全看不懂哎 前十眼。 首先 这里的logits是一个 a,b 维的张量。其中a是你的全连接输出维度,b是一个batch中的样本数量。 我们经过一个argmax的操作,dim 意味着找到张量中各自的最大值所在索引。也就是找到每个样本的全连接输出中最大的那一个。 最有可能的预测值。 torch.eq会返回一个 batch维的bool值。 。sum统 ...

2021-06-04 17:06 0 1044 推荐指数:

查看详情

torch.argmaxdim详解

torch.argmax()函数 argmax函数:torch.argmax(input, dim=None, keepdim=False)返回指定维度最大值的序号,dim给定的定义是:the demention to reduce.也就是把dim这个维度的,变成这个维度的最大值的index ...

Thu Jul 22 02:17:00 CST 2021 0 516
机器学习 | 分类性能度量指标 : ROC曲线、AUC值、正确率、召回

本篇博客的图源来自 zhwhong,转载仅作学习使用! 在分类任务中,人们总是喜欢基于错误来衡量分类器任务的成功程度。错误指的是在所有测试样例中错分的样例比例。实际上,这样的度量错误掩盖了样例如何被分错的事实。在机器学习中,有一个普遍适用的称为混淆矩阵(confusion ...

Tue Oct 26 06:29:00 CST 2021 0 305
新手学信息检索5:正确率-召回与搜索引擎的评价

这篇就简单介绍一下搜索引擎的评价方法。从用户的角度去评价一个搜索引擎的检索效果最好的方法就是计算用户在查到自己满意文档已经浏览的文档数。但是实际中,查询千变万化,文档也千变万化,所以这种方法不可行。人们便提出了下面的概念,并建立了一个评价标准。 这里面有三个常用的概念:正确率,精确,召回 ...

Wed Jan 09 19:17:00 CST 2013 0 3131
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM