How could I obtain the scores of each extracted entity?

wenwenyu / PICK-pytorch

Code for the paper "PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks" (ICPR 2020)

https://arxiv.org/abs/2004.07464

MIT License

553 stars 191 forks source link

How could I obtain the scores of each extracted entity? #84

Open jorgerodriguezsj opened 3 years ago

jorgerodriguezsj commented 3 years ago

I have investigated the scores of viterbi_decoded but it only returns in the inference the sum of everything that passes through the decoded. Would there be a way to obtain the score of each final entity extracted? thanks a lot!