ComeçarComece gratuitamente

Análise de métricas por classe

Embora as métricas agregadas sejam indicadores úteis do desempenho do modelo, muitas vezes é informativo observar as métricas por classe. Isso pode revelar classes para as quais o modelo tem desempenho inferior.

Neste exercício, você executará o loop de avaliação novamente para obter a precisão do nosso classificador de nuvem, mas, desta vez, por classe. Em seguida, você mapeará essas pontuações para os nomes das classes para interpretá-las. Como de costume, Precision já foi importado para você. Boa sorte!

Este exercício faz parte do curso

Aprendizagem profunda intermediária com PyTorch

Ver Curso

Instruções de exercício

  • Defina uma métrica de precisão apropriada para resultados por classe.
  • Calcule a precisão por classe finalizando a compreensão do dit, iterando sobre o .items() do atributo .class_to_idx de dataset_test.

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

# Define precision metric
metric_precision = Precision(
    ____, ____, ____
)

net.eval()
with torch.no_grad():
    for images, labels in dataloader_test:
        outputs = net(images)
        _, preds = torch.max(outputs, 1)
        metric_precision(preds, labels)
precision = metric_precision.compute()

# Get precision per class
precision_per_class = {
    k: ____[____].____
    for k, v 
    in ____
}
print(precision_per_class)
Editar e executar código