multi-class-text-classification-cnn-rnn icon indicating copy to clipboard operation
multi-class-text-classification-cnn-rnn copied to clipboard
verified publisher icon jiegzhan

use softmax to get probability distribution for each class

Open ctsaurabhs opened this issue 7 years ago • 0 comments

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

ctsaurabhs avatar Sep 20 '18 12:09 ctsaurabhs