Análisis de los resultados del modelo
Una vez obtenida la predicción de un modelo preentrenado, ya puedes analizarla en detalle. Tienes a tu disposición la herramienta de análisis de predicciones ( prediction
).
Verás que es una lista de longitud uno (has hecho una predicción para una sola imagen) con un diccionario con cuatro claves. « labels
» son números enteros que indican las clases predichas, y « scores
» son las probabilidades de clase correspondientes. Puedes acceder a la lista de nombres de clases correspondientes a labels
en class_names
.
¿Cuáles son las categorías previstas y sus puntuaciones asociadas para las dos máscaras principales?
Este ejercicio forma parte del curso
Aprendizaje profundo para imágenes con PyTorch
Ejercicio interactivo práctico
Pon en práctica la teoría con uno de nuestros ejercicios interactivos
