The Home Depot interview question

Code a regression model in Python?