Cálculo da precisão usando a medição da tocha
Ao programar a precisão durante o treinamento, você ajuda a identificar a época de melhor desempenho.
Neste exercício, você usará o site torchmetrics
para calcular a precisão em um conjunto de dados de máscara facial com três classes. A função plot_errors
destacará as amostras com classificação incorreta, ajudando você a analisar os erros do modelo.
torchmetrics
já está importado. O modelo outputs
são probabilidades softmax e labels
são vetores codificados com um único disparo.
Este exercício faz parte do curso
Introdução à aprendizagem profunda com PyTorch
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
outputs = model(features)
# Calculate accuracy over the batch
metric.____(____, ____)