logistic_sgd.py got to be able to return labels