Notice: This page requires JavaScript to function properly.
Please enable JavaScript in your browser settings or update your browser.
Aprende Challenge: Implementing Linear Regression | More Advanced Concepts
PyTorch Essentials

Desliza para mostrar el menú

book
Challenge: Implementing Linear Regression

Tarea

Swipe to start coding

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.

  1. Convert these columns into PyTorch tensors, and reshape them to ensure they are 2D with shapes [N, 1].
  2. Define a simple linear regression model.
  3. Use MSE as the loss function.
  4. Define optimizer as SGD with the learning rate equal to 0.01.
  5. 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.
  6. Access the model's parameters (weights and bias).

Solución

Switch to desktopCambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones
¿Todo estuvo claro?

¿Cómo podemos mejorarlo?

¡Gracias por tus comentarios!

Sección 2. Capítulo 4

Pregunte a AI

expand
ChatGPT

Pregunte lo que quiera o pruebe una de las preguntas sugeridas para comenzar nuestra charla

book
Challenge: Implementing Linear Regression

Tarea

Swipe to start coding

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.

  1. Convert these columns into PyTorch tensors, and reshape them to ensure they are 2D with shapes [N, 1].
  2. Define a simple linear regression model.
  3. Use MSE as the loss function.
  4. Define optimizer as SGD with the learning rate equal to 0.01.
  5. 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.
  6. Access the model's parameters (weights and bias).

Solución

Switch to desktopCambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones
¿Todo estuvo claro?

¿Cómo podemos mejorarlo?

¡Gracias por tus comentarios!

Sección 2. Capítulo 4
Switch to desktopCambia al escritorio para practicar en el mundo realContinúe desde donde se encuentra utilizando una de las siguientes opciones
Lamentamos que algo salió mal. ¿Qué pasó?
some-alt