Get startedGet started for free

The number of classes

To determine whether to use a binary or a multi-class model for a classification problem, you need to know the number of classes in the data. The Torch dataset is available in your workspace as train_dataset.

Considering the number of classes in train_dataset, which image classification task would be appropriate?

This exercise is part of the course

Deep Learning for Images with PyTorch

View Course

Hands-on interactive exercise

Turn theory into action with one of our interactive exercises

Start Exercise