Course Content
PyTorch Essentials
PyTorch Essentials
Challenge: Implementing Linear Regression
Task
Swipe to begin your solution
You are provided with a dataset that contains information about the number of hours students studied and their corresponding test scores. Your task is to train a linear regression model on this data.
- Convert these columns into PyTorch tensors, and reshape them to ensure they are 2D with shapes
[N, 1]
. - Define a simple linear regression model.
- Use MSE as the loss function.
- Define
optimizer
as SGD with the learning rate equal to0.01
. - Train the linear regression model to predict test scores based on the number of hours studied. At each epoch:
- Compute predictions on
X_tensor
; - Compute the loss;
- Reset the gradient;
- Perform backward pass;
- Update the parameters.
- Compute predictions on
- Access the model's parameters (weights and bias).
Solution
Switch to desktop for real-world practiceContinue from where you are using one of the options below
Everything was clear?
Thanks for your feedback!
Section 2. Chapter 4
Challenge: Implementing Linear Regression
Task
Swipe to begin your solution
You are provided with a dataset that contains information about the number of hours students studied and their corresponding test scores. Your task is to train a linear regression model on this data.
- Convert these columns into PyTorch tensors, and reshape them to ensure they are 2D with shapes
[N, 1]
. - Define a simple linear regression model.
- Use MSE as the loss function.
- Define
optimizer
as SGD with the learning rate equal to0.01
. - Train the linear regression model to predict test scores based on the number of hours studied. At each epoch:
- Compute predictions on
X_tensor
; - Compute the loss;
- Reset the gradient;
- Perform backward pass;
- Update the parameters.
- Compute predictions on
- Access the model's parameters (weights and bias).
Solution
Switch to desktop for real-world practiceContinue from where you are using one of the options below
Everything was clear?
Thanks for your feedback!
Section 2. Chapter 4
Switch to desktop for real-world practiceContinue from where you are using one of the options below