Building a Softmax Classifier for Images in PyTorch
Softmax classifier is a type of classifier in supervised learning. It is an important building block in deep learning networks and the most popular choice among deep learning practitioners.
Softmax classifier is suitable for multiclass classification, which outputs the probability for each of the classes.
This tutorial will teach you how to build a softmax classifier for images data. You will learn how to prepare the dataset, and then learn how to implement softmax classifier using PyTorch. Particularly, you’ll learn:
- About the Fashion-MNIST dataset.
- How you can use a Softmax classifier for images in PyTorch.
- How to build and train a multi-class image classifier in PyTorch.
- How to plot the results after model training.
Kick-start your project with my