Pytorch - Zero to GANs: Assignment 2 - Train Your First Model

Posted on: June 2, 2020 at 12:00 AMJune 2, 2020 at 12:00 AM (1 min read)

For this assignment we had to use personal information from a person to predict the price of medical bills for a year.

Steps

  1. Download and explore the dataset.
  2. Prepare the dataset for training.
  3. Create a linear regression model.
  4. Train the model to fit the data.
  5. Make predictions using the trained model.

Link to the Jupyter notebook I made for the assignment: remusa/02-insurance-linear-regression | Jovian