TensorFlow2学习十一之绘制准确率acc和损失函数loss曲线

model.fit中将训练集loss、测试集loss、训练集准确率保存了下来

history=model.fit(训练集数据, 训练集标签, 
	batch_size=, epochs=,
	validation_split=用作测试数据的比例,
	validation_data=测试集, 
	validation_freq=测试频率)

history包含以下几个属性:
训练集loss: loss
测试集loss: val_loss
训练集准确率: sparse_categorical_accuracy
测试集准确率: val_sparse_categorical_accuracy

acc = history.history['sparse_categorical_accuracy']
val_acc = history.history['val_sparse_categorical_accuracy']
loss = history.history['loss']
val_loss = history.history['val_loss']

绘制acc和loss曲线:

plt.subplot(1, 2, 1)
plt.plot(acc, label='Training Accuracy')
plt.plot(val_acc, label='Validation Accuracy')
plt.title('Training and Validation Accuracy')
plt.legend()

plt.subplot(1, 2, 2)
plt.plot(loss, label='Training Loss')
plt.plot(val_loss, label='Validation Loss')
plt.title('Training and Validation Loss')
plt.legend()
plt.show()
  • 12
    点赞
  • 63
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
对于使用 transformers 做 cifar10 图像分类的任务,可以使用 PyTorch 框架来实现。以下是一个简单的代码示例,包括训练模型、绘制 loss准确率曲线以及保存模型的过程: ```python import torch import torch.nn as nn import torch.optim as optim import torchvision import torchvision.transforms as transforms from transformers import ViTModel, ViTConfig # 定义模型 class ViTClassifier(nn.Module): def __init__(self): super(ViTClassifier, self).__init__() config = ViTConfig(image_size=32, num_classes=10) self.vit = ViTModel(config) self.classifier = nn.Linear(config.hidden_size, config.num_classes) def forward(self, x): x = self.vit(x) x = self.classifier(x.last_hidden_state[:, 0]) return x # 数据预处理 transform_train = transforms.Compose([ transforms.RandomCrop(32, padding=4), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) transform_test = 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_train) trainloader = torch.utils.data.DataLoader(trainset, batch_size=128, shuffle=True, num_workers=2) testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform_test) testloader = torch.utils.data.DataLoader(testset, batch_size=100, shuffle=False, num_workers=2) # 定义损失函数和优化器 criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(net.parameters(), lr=0.1, momentum=0.9, weight_decay=5e-4) # 开始训练 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") net = ViTClassifier().to(device) num_epochs = 50 loss_list = [] acc_list = [] for epoch in range(num_epochs): running_loss = 0.0 total = 0 correct = 0 for i, data in enumerate(trainloader, 0): inputs, labels = data inputs, labels = inputs.to(device), labels.to(device) 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() loss_list.append(running_loss / len(trainloader)) acc_list.append(correct / total) print('[Epoch %d] Loss: %.3f | Acc: %.3f' % (epoch + 1, running_loss / len(trainloader), correct / total)) # 绘制 loss准确率曲线 import matplotlib.pyplot as plt plt.plot(loss_list) plt.xlabel('Epoch') plt.ylabel('Loss') plt.show() plt.plot(acc_list) plt.xlabel('Epoch') plt.ylabel('Accuracy') plt.show() # 保存模型 torch.save(net.state_dict(), 'model.pth') ``` 在训练过程中,我们记录了每个 epoch 的 loss准确率,并将它们存储在 `loss_list` 和 `acc_list` 中。然后使用 Matplotlib 库绘制出了相应的 loss准确率曲线。最后,我们使用 `torch.save()` 函数将模型保存到磁盘中,以便后续的推理和部署。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

泤燃

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值