请在代码中加入准确率:with torch.no_grad(): class_accuary_List=[] true_labels = [] predicted_labels = [] for data in testloader: images, labels = data images = images.view(len(images), 1, 121).float() inputs, labels = images.to(device), labels.to(device) outputs = cnn(inputs) _, predicted = torch.max(outputs, 1) true_labels.append(labels.cpu().numpy()) predicted_labels.append(predicted.cpu().numpy()) true_labels = np.concatenate(true_labels) predicted_labels = np.concatenate(predicted_labels) cm = confusion_matrix(true_labels, predicted_labels, labels=[0, 1, 2, 3, 4]) print(cm) for i in range(cm.shape[0]): acc = cm[i, i] / np.sum(cm[i]) class_accuary_List.append(acc) print("Class", i, "Accuracy:", acc)

时间: 2024-02-22 16:56:03 浏览: 24
好的,下面是加入准确率的代码: ``` with torch.no_grad(): class_accuracy_list = [] true_labels = [] predicted_labels = [] for data in testloader: images, labels = data images = images.view(len(images), 1, 121).float() inputs, labels = images.to(device), labels.to(device) outputs = cnn(inputs) _, predicted = torch.max(outputs, 1) true_labels.append(labels.cpu().numpy()) predicted_labels.append(predicted.cpu().numpy()) true_labels = np.concatenate(true_labels) predicted_labels = np.concatenate(predicted_labels) cm = confusion_matrix(true_labels, predicted_labels, labels=[0, 1, 2, 3, 4]) print(cm) for i in range(cm.shape[0]): acc = cm[i, i] / np.sum(cm[i]) class_accuracy_list.append(acc) print("Class", i, "Accuracy:", acc) overall_acc = np.sum(np.diag(cm)) / np.sum(cm) print("Overall Accuracy:", overall_acc) ``` 这里加入了一个 `overall_acc` 变量,用于计算整体准确率。同时,在循环中计算每个类别的准确率,并将其存储在 `class_accuracy_list` 列表中。最后输出整体准确率和每个类别的准确率。

相关推荐

详细解释代码import torch import torch.nn as nn import torch.optim as optim import torchvision import torchvision.transforms as transforms from torch.utils.data import DataLoader # 图像预处理 transform = transforms.Compose( [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) # 加载数据集 trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) trainloader = DataLoader(trainset, batch_size=128, shuffle=True, num_workers=0) testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform) testloader = DataLoader(testset, batch_size=128, shuffle=False, num_workers=0) # 构建模型 class RNNModel(nn.Module): def init(self): super(RNNModel, self).init() self.rnn = nn.RNN(input_size=3072, hidden_size=512, num_layers=2, batch_first=True) self.fc = nn.Linear(512, 10) def forward(self, x): # 将输入数据reshape成(batch_size, seq_len, feature_dim) x = x.view(-1, 3072, 1).transpose(1, 2) x, _ = self.rnn(x) x = x[:, -1, :] x = self.fc(x) return x net = RNNModel() # 定义损失函数和优化器 criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(net.parameters(), lr=0.001) # 训练模型 loss_list = [] acc_list = [] for epoch in range(30): # 多批次循环 running_loss = 0.0 correct = 0 total = 0 for i, data in enumerate(trainloader, 0): # 获取输入 inputs, labels = data # 梯度清零 optimizer.zero_grad() # 前向传播,反向传播,优化 outputs = net(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() # 打印统计信息 running_loss += loss.item() _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() acc = 100 * correct / total acc_list.append(acc) loss_list.append(running_loss / len(trainloader)) print('[%d] loss: %.3f, acc: %.3f' % (epoch + 1, running_loss / len(trainloader), acc)) print('Finished Training') torch.save(net.state_dict(), 'rnn1.pt') # 绘制loss变化曲线和准确率变化曲线 import matplotlib.pyplot as plt fig, axs = plt.subplots(2, 1, figsize=(10, 10)) axs[0].plot(loss_list) axs[0].set_title("Training Loss") axs[0].set_xlabel("Epoch") axs[0].set_ylabel("Loss") axs[1].plot(acc_list) axs[1].set_title("Training Accuracy") axs[1].set_xlabel("Epoch") axs[1].set_ylabel("Accuracy") plt.show() # 测试模型 correct = 0 total = 0 with torch.no_grad(): for data in testloader: images, labels = data outputs = net(images) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() print('Accuracy of the network on the 10000 test images: %d %%' % (100 * correct / total))

最新推荐

recommend-type

前端Mock的使用,用于构造动态数据

适用于前端开发,前端构建动态数据
recommend-type

SQL语句的基本用法案例.pdf

“SQL语句的基本用法案例”的文档,是一份非常实用的学习资料,为初学者和进阶者提供了丰富的SQL操作示例。通过这份文档,读者可以系统地了解SQL语言在数据库管理中的应用,掌握从创建数据库到删除数据库的整个流程。 文档开篇便介绍了如何创建一个新的数据库,并详细说明了选择数据库、创建表以及插入数据的具体步骤。这不仅为读者展示了SQL语句的基础用法,也为后续的查询、更新和删除操作打下了坚实的基础。 在查询数据部分,文档通过多个示例展示了SQL查询的灵活性和强大功能。无论是查询所有学生信息,还是根据特定条件筛选数据,都能通过简单的SQL语句实现。此外,文档还介绍了如何计算学生的总数、平均年龄等统计信息,以及如何对数据进行排序和限制结果。 除了基础的增删改查操作,文档还深入介绍了子查询、连接表以及窗口函数等高级用法。这些功能在实际应用中非常常见,能够帮助用户解决更为复杂的数据处理问题。 此外,文档还提供了清晰的代码示例和注释,使得读者能够轻松理解每个步骤的含义和目的。同时,文档的语言简洁明了,逻辑清晰,使得学习过程更加轻松愉快。
recommend-type

RTL8188FU-Linux-v5.7.4.2-36687.20200602.tar(20765).gz

REALTEK 8188FTV 8188eus 8188etv linux驱动程序稳定版本, 支持AP,STA 以及AP+STA 共存模式。 稳定支持linux4.0以上内核。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

:YOLOv1目标检测算法:实时目标检测的先驱,开启计算机视觉新篇章

![:YOLOv1目标检测算法:实时目标检测的先驱,开启计算机视觉新篇章](https://img-blog.csdnimg.cn/img_convert/69b98e1a619b1bb3c59cf98f4e397cd2.png) # 1. 目标检测算法概述 目标检测算法是一种计算机视觉技术,用于识别和定位图像或视频中的对象。它在各种应用中至关重要,例如自动驾驶、视频监控和医疗诊断。 目标检测算法通常分为两类:两阶段算法和单阶段算法。两阶段算法,如 R-CNN 和 Fast R-CNN,首先生成候选区域,然后对每个区域进行分类和边界框回归。单阶段算法,如 YOLO 和 SSD,一次性执行检
recommend-type

设计算法实现将单链表中数据逆置后输出。用C语言代码

如下所示: ```c #include <stdio.h> #include <stdlib.h> // 定义单链表节点结构体 struct node { int data; struct node *next; }; // 定义单链表逆置函数 struct node* reverse(struct node *head) { struct node *prev = NULL; struct node *curr = head; struct node *next; while (curr != NULL) { next
recommend-type

c++校园超市商品信息管理系统课程设计说明书(含源代码) (2).pdf

校园超市商品信息管理系统课程设计旨在帮助学生深入理解程序设计的基础知识,同时锻炼他们的实际操作能力。通过设计和实现一个校园超市商品信息管理系统,学生掌握了如何利用计算机科学与技术知识解决实际问题的能力。在课程设计过程中,学生需要对超市商品和销售员的关系进行有效管理,使系统功能更全面、实用,从而提高用户体验和便利性。 学生在课程设计过程中展现了积极的学习态度和纪律,没有缺勤情况,演示过程流畅且作品具有很强的使用价值。设计报告完整详细,展现了对问题的深入思考和解决能力。在答辩环节中,学生能够自信地回答问题,展示出扎实的专业知识和逻辑思维能力。教师对学生的表现予以肯定,认为学生在课程设计中表现出色,值得称赞。 整个课程设计过程包括平时成绩、报告成绩和演示与答辩成绩三个部分,其中平时表现占比20%,报告成绩占比40%,演示与答辩成绩占比40%。通过这三个部分的综合评定,最终为学生总成绩提供参考。总评分以百分制计算,全面评估学生在课程设计中的各项表现,最终为学生提供综合评价和反馈意见。 通过校园超市商品信息管理系统课程设计,学生不仅提升了对程序设计基础知识的理解与应用能力,同时也增强了团队协作和沟通能力。这一过程旨在培养学生综合运用技术解决问题的能力,为其未来的专业发展打下坚实基础。学生在进行校园超市商品信息管理系统课程设计过程中,不仅获得了理论知识的提升,同时也锻炼了实践能力和创新思维,为其未来的职业发展奠定了坚实基础。 校园超市商品信息管理系统课程设计的目的在于促进学生对程序设计基础知识的深入理解与掌握,同时培养学生解决实际问题的能力。通过对系统功能和用户需求的全面考量,学生设计了一个实用、高效的校园超市商品信息管理系统,为用户提供了更便捷、更高效的管理和使用体验。 综上所述,校园超市商品信息管理系统课程设计是一项旨在提升学生综合能力和实践技能的重要教学活动。通过此次设计,学生不仅深化了对程序设计基础知识的理解,还培养了解决实际问题的能力和团队合作精神。这一过程将为学生未来的专业发展提供坚实基础,使其在实际工作中能够胜任更多挑战。
recommend-type

"互动学习:行动中的多样性与论文攻读经历"

多样性她- 事实上SCI NCES你的时间表ECOLEDO C Tora SC和NCESPOUR l’Ingén学习互动,互动学习以行动为中心的强化学习学会互动,互动学习,以行动为中心的强化学习计算机科学博士论文于2021年9月28日在Villeneuve d'Asq公开支持马修·瑟林评审团主席法布里斯·勒菲弗尔阿维尼翁大学教授论文指导奥利维尔·皮耶昆谷歌研究教授:智囊团论文联合主任菲利普·普雷教授,大学。里尔/CRISTAL/因里亚报告员奥利维耶·西格德索邦大学报告员卢多维奇·德诺耶教授,Facebook /索邦大学审查员越南圣迈IMT Atlantic高级讲师邀请弗洛里安·斯特鲁布博士,Deepmind对于那些及时看到自己错误的人...3谢谢你首先,我要感谢我的两位博士生导师Olivier和Philippe。奥利维尔,"站在巨人的肩膀上"这句话对你来说完全有意义了。从科学上讲,你知道在这篇论文的(许多)错误中,你是我可以依
recommend-type

:YOLO目标检测算法的挑战与机遇:数据质量、计算资源与算法优化,探索未来发展方向

![:YOLO目标检测算法的挑战与机遇:数据质量、计算资源与算法优化,探索未来发展方向](https://img-blog.csdnimg.cn/7e3d12895feb4651b9748135c91e0f1a.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBA5rKJ6YaJ77yM5LqO6aOO5Lit,size_20,color_FFFFFF,t_70,g_se,x_16) # 1. YOLO目标检测算法简介 YOLO(You Only Look Once)是一种
recommend-type

在xlwings里.循环读取A列和B列的数据。当A列某单元格值等于特品,且同行B列单元格值等于热销品时。在同行的C列单元格输出值999。这代码怎么写

您好!感谢您对ChitGPT的使用! 为了回答您的问题,您可以使用xlwings的API和Excel公式配合完成以下任务: ``` import xlwings as xw wb = xw.Book('workbook.xlsx') # 打开工作簿 sht = wb.sheets['Sheet1'] # 打开工作表 # 遍历A列和B列数据 for i in range(1, sht.range('A' + str(sht.cells.last_cell.row)).end('up').row + 1): if sht.range(f'A{i}').value == '特品'