Sign In

Assignment 2 - Train Your First Model

Deep Learning with PyTorch: Zero to GANs

In this assignment, you’re going to use information like a person’s age, sex, BMI, no. of children, and smoking habits to predict the price of yearly medical bills. You will train a model with the following steps:

  • Download and explore the dataset
  • Prepare the dataset for training
  • Create a linear regression model
  • Train the model to fit the data
  • Make predictions using the trained model