This approach assumes that you use BCELoss() ("binary cross entropy loss") during training. The underlying theory and mechanics of output activation and loss function are complex, but you don't need ...

The Data Science Lab. Binary Classification Using New PyTorch Best Practices, Part 2: Training, Accuracy, Predictions. Dr. James McCaffrey of Microsoft Research explains how to train a network, ...