在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
投稿猜你喜欢
2007年1月，国务院 * 了中央“一号文件”，文件中对加快农业信息化建设有了更明确的部署，为新农村建- 直接上代码：#!/usr/bin/python# -*- coding: utf-8 -*-import urllibimport os,d
- 本文讲述了python开发之IDEL(Python GUI)的使用方法。分享给大家供大家参考，具体如下：在安装完Python后，我们希望能够
- 前言通常执行 python 程序要有相应的 Python 环境，但某些特定场景下，我们可能并不愿意这么麻烦的去配置这些环境（比如将写好的脚本
- 一、PIL库对图像的基本操作1、读取图片PIL网上有很多介绍，这里不再讲解。直接操作，读取一张图片，将其转换为灰度图像，并打印出来。from
- 本文实例讲述了php使用curl获取https请求的方法。分享给大家供大家参考。具体分析如下：今日在做一个项目，需要curl获取第三方的AP
- php cookie中不能使用点号（句号），实际上不是很严格，应该说可以使用点号的cookie名，但会被转换，你命名一个cookie:$_C
- 如果是感应触发.就选onmouseover如果是点击触发.就选onclick [把它们两互相替换,就可随时变为感应
- 下面给大家介绍下pandas读取CSV文件时查看修改各列的数据类型格式，具体内容如下所述：我们在调bug的时候会经常查看、修改pandas列
- 所谓Julia集就是类似下面的美妙的图案Julia集特别地，当 c = z的初始值时，符合收敛条件的 z 的便构成大名鼎鼎的Mandelbr
- 简单的仿图片验证码，适合新手简单的仿图片验证码演示，很容易被破解，实用性不大，但拿出来给新手学习一下还是不错的：JScript.Asp代码示
- Mcrypt扩展库可以实现加密解密功能，就是既能将明文加密，也可以密文还原。1.PHP加密扩展库Mcrypt安装在标准的PHP安装过程中并没
- 本文实例讲述了Python利用前序和中序遍历结果重建二叉树的方法。分享给大家供大家参考，具体如下：题目：输入某二叉树的前序遍历和中序遍历的结
- PDO::commitPDO::commit提交一个事务(PHP 5 >= 5.1.0, PECL pdo >= 0.1.0)说
- Python 是由吉多·范罗苏姆(Guido Van Rossum)在 90 年代早期设计。 它是如今最常用的编程语言之一。它的语法简洁且优
- 您是否常常在做网页的过程中发现一个问题呢？当图片上传的时候，如果图片太大 ，就会把网页撑破，唯一做的就要先把它用软件缩小，再上传上
- exec函数，可以循环定义、赋值多个变量exec ("temp％s=1"％1)这段代码的意思是，让exec执行temp1
- ASP链接MSSQL2005的链接字符串如下：Provider=SQLNCLI;Server=.\SQLEXPRESS;Database=m
- 本文实例讲述了python生成器用法。分享给大家供大家参考，具体如下：1. 生成器利用迭代器，我们可以在每次迭代获取数据（通过next()方
- 今天在 ajaxian 上看到一篇文章，名为 Five Ajax Anti-pattern ，觉得讲得比较有道理，现粗略翻译一下，加一些自己
