El número de clases

Para determinar si se debe utilizar un modelo binario o multiclase para un problema de clasificación, es necesario conocer el número de clases que hay en los datos. El conjunto de datos Torch está disponible en tu espacio de trabajo como train_dataset.

Teniendo en cuenta el número de clases en train_dataset, ¿qué tarea de clasificación de imágenes sería la más adecuada?