Softmax Classifier Optimization via Gradient Descent - Coding problem | OneNoughtOne