Deep Learning with PyTorch Quick Start Guide

Logistic regression

We can use the linear regression model to perform binary classification by finding a decision boundary that divides two predicted classes. A common way to do this is by using a sigmoid function, defined as the following:

The plot of the sigmoid function looks like this:

The sigmoid function can be used in the hypothesis function, to output a probability, as follows:

Here, the output of the hypothesis function is the probability ...