Contenido del Curso
Neural Networks with PyTorch
Neural Networks with PyTorch
Challenge: Implementing Linear Regression
Tarea
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).
Solución
Cambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones
¿Todo estuvo claro?
¡Gracias por tus comentarios!
Sección 2. Capítulo 4
Challenge: Implementing Linear Regression
Tarea
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).
Solución
Cambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones
¿Todo estuvo claro?
¡Gracias por tus comentarios!
Sección 2. Capítulo 4
Cambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones