use softmax to get probability distribution for each class

jiegzhan / multi-class-text-classification-cnn-rnn

Classify Kaggle San Francisco Crime Description into 39 classes. Build the model with CNN, RNN (GRU and LSTM) and Word Embeddings on Tensorflow.

https://www.kaggle.com/c/sf-crime/data

Apache License 2.0

599 stars 262 forks source link

use softmax to get probability distribution for each class #38

Open ctsaurabhs opened 6 years ago

ctsaurabhs commented 6 years ago

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