Regression Model Python Code