use softmax to get probability distribution for each class

[ctsaurabhs]

I wish to get raw probabilities of each class, instead of argmax() returning the index for highest probability.
request you to kindly help me in using softmax() in the predict.py. Thanks