How to get confidence for each token in prediction

monologg / JointBERT

Pytorch implementation of JointBERT: "BERT for Joint Intent Classification and Slot Filling"

Apache License 2.0

662 stars 186 forks source link

How to get confidence for each token in prediction #11

Open 008karan opened 4 years ago

008karan commented 4 years ago

first of all good job on this repo @monologg ! I facing difficulty in getting confidence of each predicted entity. I am able to get confidence for each intent. Not able to figure out how to parse confidence of each predicted entity. Can you help here?