Analyse des résultats du modèle
prediction
Une fois que vous avez obtenu la prédiction à partir d'un modèle pré-entraîné, vous pouvez maintenant l'analyser en détail. La fonctionnalité « Analyse de prédiction » est à votre disposition.
Vous constaterez qu'il s'agit d'une liste d'une longueur de un (vous n'avez fait de prédiction que pour une seule image) avec un dictionnaire contenant quatre clés. labels
sont des nombres entiers indiquant les classes prédites, et scores
sont les probabilités de classe correspondantes. Vous pouvez accéder à la liste des noms de classes correspondant à labels
à l'adresse class_names
.
Quelles sont les catégories prévues et les scores associés pour les deux masques les mieux classés ?
Cet exercice fait partie du cours
Deep learning pour les images avec PyTorch
Exercice interactif pratique
Passez de la théorie à la pratique avec l’un de nos exercices interactifs
