Softmax Regression (multi-task L2 norm Logistic Regression) in python