Tutorial 2: Train a linear regression model using the SGD optimizer