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
Hands-on interactive exercise
Turn theory into action with one of our interactive exercises
