Cálculo da precisão usando torchmetrics
O monitoramento da precisão durante o treinamento ajuda a identificar a época com melhor desempenho.
Neste exercício, você deve usar torchmetrics
para calcular a precisão em um conjunto de dados sobre uso de máscara facial com três classes. A função plot_errors
destaca as amostras com classificação incorreta, ajudando você a analisar os erros do modelo.
torchmetrics
já foi importado. As saídas (outputs
) do modelo são probabilidades softmax, e os rótulos (labels
) são vetores com one-hot encoding.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
outputs = model(features)
# Calculate accuracy over the batch
metric.____(____, ____)