召回率R,精确率P,精度accuracy,F1计算

一、召回率R,精确率P,精度accuracy,F1定义

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

二、使用步骤

from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score

#1表示白球,0表示黑球
y_true = [1, 1, 1, 0, 0, 0, 1, 0, 0, 1]
y_pred = [1, 1, 0, 1, 1, 0, 1, 0, 0, 1]

print("acc:", accuracy_score(y_true, y_pred))
print("p:", precision_score(y_true, y_pred))
print("r:", recall_score(y_true, y_pred))
print("f1:", f1_score(y_true, y_pred))

打印输出
在这里插入图片描述
在这里插入图片描述
例如,图中白球为1,黑球为0
TP = 4,实际的白球,被预测为白球,编号1,2,7,10
FP = 2,实际的黑球,被预测为白球,编号4,5
FN = 1,实际的白球,被预测为黑球,编号3
TN = 3,实际的黑球,被预测为黑球,编号6,8,9

所以
Precision = 4/(4+2) = 2/3 = 0.66667
Recall = 4/(4+1) = 0.8
acc = (4+3)/10 = 0.7
在这里插入图片描述

参考

1.https://blog.csdn.net/lhxez6868/article/details/108150777

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
在PyTorch中,可以使用一些函数和类来计算精度、回归和F1 score等指标。 对于分类问题中的精度,可以通过以下步骤来计算: 1. 首先,需要将模型预测的结果转化为类别标签。可以使用torch.argmax()函数来获取每个样本预测概最大的类别。 2. 然后,将预测的类别与真实标签进行比较,以确定模型预测的准确性。可以使用torch.eq()函数来判断预测结果是否与真实标签相等。 3. 最后,计算预测准确的样本数占总样本数的比例,即精度。可以使用torch.mean()函数来计算平均精度。 以下是一个示例代码片段,演示了如何计算精度: ```python # 导入必要的模块 import torch import torch.nn.functional as F # 假设有模型输出和真实标签 outputs = torch.tensor([[0.2, 0.8], [0.6, 0.4], [0.3, 0.7]]) labels = torch.tensor([1, 0, 1]) # 获取预测类别 predicted_labels = torch.argmax(outputs, dim=1) # 计算精度 accuracy = torch.mean(torch.eq(predicted_labels, labels).float()) print(accuracy.item()) # 输出精度值 ``` 对于二分类问题中的回归,可以使用混淆矩阵来计算。可以使用Sklearn库中的metrics模块来计算回归。 以下是一个示例代码片段,演示了如何使用混淆矩阵计算回归: ```python from sklearn.metrics import confusion_matrix, recall_score # 假设有模型输出和真实标签 outputs = torch.tensor([1, 0, 1, 1]) labels = torch.tensor([1, 0, 0, 1]) # 获取预测类别 predicted_labels = torch.round(torch.sigmoid(outputs)).int() # 计算混淆矩阵 conf_matrix = confusion_matrix(labels.numpy(), predicted_labels.numpy()) # 计算回归 recall = recall_score(labels.numpy(), predicted_labels.numpy()) print(conf_matrix) # 输出混淆矩阵 print(recall) # 输出回归 ``` F1 score是精确召回的加权平均,可以使用Sklearn库中的metrics模块来计算。 以下是一个示例代码片段,演示了如何计算F1 score: ```python from sklearn.metrics import f1_score # 假设有模型输出和真实标签 outputs = torch.tensor([0.2, 0.7, 0.3, 0.8]) labels = torch.tensor([0, 1, 1, 1]) # 获取预测类别 predicted_labels = torch.round(torch.sigmoid(outputs)).int() # 计算F1 score f1 = f1_score(labels.numpy(), predicted_labels.numpy()) print(f1) # 输出F1 score ``` 以上示例代码演示了在PyTorch中计算精度、回归和F1 score等指标的方法。根据具体问题的要求和使用的库可以适当进行调整和修改。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值