Linear regression 1D: Training Two Parameter Stochastic Gradient Descent (SGD)

Objective

• How to use SGD(Stochastic Gradient Descent) to train the model.

Table of Contents

In this Lab, you will practice training a model by using Stochastic Gradient descent.

• Make Some Data
• Create the Model and Cost Function (Total Loss)
• Train the Model:Batch Gradient Descent
• Train the Model:Stochastic gradient descent
• Train the Model:Stochastic gradient descent with Data Loader

Estimated Time Needed: 30 min