Model Types

Nested models

In our model, we manually created two parameters to perform a linear regression. Instead of defining individual parameters, what if we use PyTorch’s Linear model?

We are implementing a single feature linear regression, one input, and one output, so the corresponding linear model would look like this: