Gradient Descent Algorithm Python Code