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
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
dedataset_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)