Next, the demo uses the trained model to make a prediction ... shown in Listing 4 demonstrates how to train a network for multi-class classification. The screenshot in Figure 2 shows the output from ...

For an example, see my "PyTorch Multi-Class Accuracy by Class" blog post. Using the Model After the ... that are specific to binary classification. For example, sigmoid() activation rather than ...