Analysera modellens utdata
Nu när du har fått en prediktion från en förtränad modell kan du analysera den i detalj. prediction finns tillgänglig för dig.
Du kommer att se att det är en lista med längden ett (du gjorde en prediktion för bara en bild) med en ordbok med fyra nycklar. labels är heltal som anger de förutsagda klasserna, och scores är motsvarande klassprobabiliteter. Du kan komma åt listan med klassnamn som motsvarar labels via class_names.
Vilka är de förutsagda kategorierna och deras tillhörande poäng för de två bästa maskerna?
Den här övningen är en del av kursen
Djupinlärning för bilder med PyTorch
Interaktiv övning med praktiskt arbete
Gör teori till handling med en av våra interaktiva övningar
Starta övningen