Train Softmax Regression with Gradient DescentDeep-ML | Problem