ComeçarComece de graça

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

Ver curso

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.____(____, ____)
Editar e executar o código