ComeçarComece gratuitamente

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

Ver Curso

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