Analiza rezultatelor modelului
După ce ai obținut predicția de la un model pre-antrenat, o poți analiza în detaliu. Variabila prediction îți este disponibilă.
Vei observa că este o listă cu un singur element (ai făcut predicția pentru o singură imagine), care conține un dicționar cu patru chei. labels sunt numere întregi ce indică clasele prezise, iar scores sunt probabilitățile corespunzătoare fiecărei clase. Lista numelor de clase corespunzătoare valorilor din labels este disponibilă ca class_names.
Care sunt categoriile prezise și scorurile asociate pentru primele două măști?
Acest exercițiu face parte din cursul
Deep Learning pentru imagini cu PyTorch
Exercițiu interactiv practic
Transformă teoria în practică cu unul dintre exercițiile noastre interactive
Începe exercițiul