1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

道练习

使用 torchmetrics 计算准确率

在训练过程中跟踪准确率有助于找出表现最好的 epoch。

在本练习中，您将使用 torchmetrics 来计算一个包含 3 个类别的口罩数据集的准确率。plot_errors 函数会高亮标出被误分类的样本，帮助您分析模型错误。

torchmetrics 包已导入。模型的 outputs 是 softmax 概率，labels 是独热编码向量。

说明 1 / 共 2 个

undefined XP
    1
    2
  • 创建一个 "multiclass" 的准确率指标，类别数为3。
  • 计算数据加载器中每个批次的准确率。