使用 torchmetrics 计算准确率
在训练过程中跟踪准确率有助于找出表现最好的 epoch。
在本练习中，您将使用 torchmetrics 来计算一个包含 3 个类别的口罩数据集的准确率。plot_errors 函数会高亮标出被误分类的样本，帮助您分析模型错误。
torchmetrics 包已导入。模型的 outputs 是 softmax 概率，labels 是独热编码向量。
本练习是课程的一部分
使用 PyTorch 的深度学习入门
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
outputs = model(features)
# Calculate accuracy over the batch
metric.____(____, ____)