search

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
559 stars 193 forks source link

Is it possible to get score in the form of probabilities? #28

Closed Vcv85 closed 4 years ago

Vcv85 commented 4 years ago

Hi wenwenyu,

Is it possible to get score in the form of probabilities?

wenwenyu commented 4 years ago

The viterbi decode will produce the best path and its scores. The output of scores can be regarded as probability. But it truly is the cost of transitions and logit. Details of the implementation of viterbi decode can refer to viterbi_decode.

Vcv85 commented 4 years ago

Thank you. I will look