使用 torchmetrics 計算 accuracy
在訓練過程中追蹤 accuracy 有助於找出效能最佳的 epoch。
在這個練習中，你將使用 torchmetrics 來計算具有三個類別的口罩資料集的 accuracy。plot_errors 函式會標示被錯分的樣本，幫助你分析模型的錯誤。
torchmetrics 套件已匯入。模型的 outputs 是 softmax 機率，labels 則是 one-hot 編碼向量。
本練習屬於課程
使用 PyTorch 的深度學習入門
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
outputs = model(features)
# Calculate accuracy over the batch
metric.____(____, ____)