Introduction to PyTorch: from training loop to prediction

That said, let's see what the code for writing a logistic regression model looks like. Our class inherits from nn.Module. This class provides the methods behind the scenes that make the model work. The __init__ method of a class contains the logic that runs when instantiating a class in Python. Here we pass two arguments: the number of features and the number of classes to predict.