Katkat
Back

Implement Batch Gradient Descent for Linear Regression

AmazonData Scientist · Technical Screen
EasyData ScientistTechnical ScreenMachine Learning

Last updated

Amazon data scientist machine learning prompt on writing batch gradient descent pseudocode for linear regression, including MSE loss, vectorized gradients, parameter updates, stopping criteria, and implementation checks.

Scenario

Building a linear regression model from scratch; parameters are optimized through batch gradient descent.

Question

Write Python-style pseudocode for batch gradient descent that minimizes mean-squared error for linear regression. Explain briefly what each step in your pseudocode does.

Hints

Cover initialization, prediction, error, gradient, parameter update, loop/stop conditions.