PyTorch-for-Linear-Regression

  • Load data and create a train/test split.
  • Build your own Pytorch model for simple linear regression problem.
  • Train the model with gradient descent algorithm in Pytorch.