stochastic gradient descent python code