The Home Depot Interview Question

Code a regression model in Python?