Stochastic Gradient Descent Python