Extract Associated Probability

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

Extract Associated Probability #24

Closed morgan-hd closed 7 years ago

morgan-hd commented 7 years ago

This is awesome code. I am fairly new to TF and have tried to google the answer myself but can't figure it out. How do I also extract the associated probability with the label that it predicts?