Gradient Descent Regression Python