Multi-Class Classification Using PyTorch: Defining a Network
Dr. James McCaffrey of Microsoft Research explains how to define a network in installment No. 2 of his four-part series that will present a complete end-to-end production-quality example of multi-class classification using a PyTorch neural network.
12/15/2020 The goal of a multi-class classification problem is to predict a value that can be one of three or more possible discrete values, such as red, yellow or green for a traffic signal. This article is the second in a series of four articles that present a complete end-to-end production-quality example of multi-class classification using a PyTorch neural network. The example problem is to predict a college student s major ( finance, geology or history ) from their sex, number of units completed, home state and score on an admission test.