在pytorch 中计算精度、回归率、F1 score等指标的实例
作者：Link2Link 发布时间：2022-08-10 06:28:18
标签：pytorch,精度,回归率,F1,score
pytorch中训练完网络后，需要对学习的结果进行测试。官网上例程用的方法统统都是正确率，使用的是torch.eq()这个函数。
但是为了更精细的评价结果，我们还需要计算其他各个指标。在把官网API翻了一遍之后发现并没有用于计算TP，TN，FP，FN的函数。。。
在动了无数歪脑筋之后，心想pytorch完全支持numpy，那能不能直接进行判断，试了一下果然可以，上代码：
# TP predict 和 label 同时为1
TP += ((pred_choice == 1) & (target.data == 1)).cpu().sum()
# TN predict 和 label 同时为0
TN += ((pred_choice == 0) & (target.data == 0)).cpu().sum()
# FN predict 0 label 1
FN += ((pred_choice == 0) & (target.data == 1)).cpu().sum()
# FP predict 1 label 0
FP += ((pred_choice == 1) & (target.data == 0)).cpu().sum()
p = TP / (TP + FP)
r = TP / (TP + FN)
F1 = 2 * r * p / (r + p)
acc = (TP + TN) / (TP + TN + FP + FN
这样就能看到各个指标了。
因为target是Variable所以需要用target.data取到对应的tensor，又因为是在gpu上算的，需要用 .cpu() 移到cpu上。
因为这是一个batch的统计，所以需要用+=累计出整个epoch的统计。当然，在epoch开始之前需要清零
来源：https://blog.csdn.net/qq_15602569/article/details/79565402
0
投稿猜你喜欢
- 代码如下: Function NumberSplit(num) Dim i,length length=Len(num) For i=1 T
数据集数据集为Barcelona某段时间内的气象数据，其中包括温度、湿度以及风速等。本文将简单搭建来对风速进行预测。特征构造对于风速的预测，- 数据库系统是管理信息系统的核心，基于数据库的联机事务处理(OLTP)以及联机分析处理(OLAP)是银行、企业、政府等部门最为重要的计算机应用
- 大家知道，mailto是网页设计制作中的一个非常实用的html标签，许多拥有个人网页的朋友都喜欢在网站的醒目位置处写上自己的电子邮件地址，这
- 问题如图：想要删除产品表的主键约束，但是报错创建的产品表：CREATE TABLE PROVIDERS(PROVIDERID INT NOT
- 遇到的问题网上找了一些代码，都是只能建立一次socket传输一张图片，然后断开重新连重新传。而建立一次socket代价不小，反复建立会非常消
- CHAR和VARCHAR类型相似，差别主要在存储，尾随空格和检索方式上。CHAR和VARCHAR相同的是：CHAR和VARCHAR都指定了字
- 一、什么是frame&frame切换？frame：HTML页面中的一种框架，主要作用是在当前页面中指定区域显示另一页面元素；fram
- 前言：集合这个词应该比较耳熟，大多数人没接触代码前就学过了。回想一下你的高一数学课本上是不是出现过这个词，就在第一章，概念如下：一般地，我们
- 展示import tkinter if __name__ == '__main__': win =
- pytorch更新完后合并了Variable与Tensortorch.Tensor()能像Variable一样进行反向传播的更新，返回值为T
- 下面附上参考文章，这篇文章是通过识别出来的文字来打开浏览器中的默认网站。python通过调用百度api实现语音识别题目很简单，利用语音识别识
- 本文实例讲述了Python读取一个目录下所有目录和文件的方法。分享给大家供大家参考，具体如下：这里介绍的是刚学python时的一个读取目录的
- PHP 5.0.0 和PHP 4.0.38 于2004年7月13日同时发布，这是一个值得我们PHP爱好者的一大喜讯。期盼已久的PHP5终于出
- 感觉这个东西很常用，很多其它js框架里有，但是太大了，用不了太多，自己写了个。需要的朋友可以参考下。下面是html测试代码：<!DOC
- WordPress 3.0 引入导航菜单功能, 让页面的导航和链接的管理变得简单易用. WP 向用户提供了菜单管理页面和多种调用方法, 我们
- using System; using System.Data; using System.Configuration; using Sys
- 大名鼎鼎的FCKeditor终于在最近发布新版本了，与增加版本号不同，这次完全把它改名了，更名为CKeditor。这应该是和它的开发公司CK
- 本文实例讲述了python抓取百度首页的方法。分享给大家供大家参考。具体实现方法如下：import urllibdef downURL(ur
- 1.MySQL中并发和隔离控制机制Meta-data元数据锁：在table cache缓存里实现的，为DDL(Data Definition
