Количество классов

Чтобы выбрать подходящую модель для задачи классификации — бинарную или многоклассовую, — нужно знать количество классов в данных. Набор данных Torch доступен в вашем рабочем пространстве под именем train_dataset.

Принимая во внимание количество классов в train_dataset, какая задача классификации изображений подойдёт в данном случае?