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
553 stars 191 forks source link

Confidence threshold of predictions #74

Open compadrejavo opened 3 years ago

compadrejavo commented 3 years ago

Greetings.

First of all, thanks for your work on this project; second, after training a model and obtaining the predictions, it only shows the predictions that have a very high confidence level (0.9999988675117493) and the rest simply don't appear, so I was wondering if there is a threshold or something in the code that I can tweak to see predictions with lower confidence (like 0.95 or 0.90).

Thanks in advance.

jorgerodriguezsj commented 3 years ago

I would also like to achieve this, if you have any progress please tell me.

Thanks!