PICK-pytorch
PICK-pytorch copied to clipboard
Confidence threshold of predictions
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.
I would also like to achieve this, if you have any progress please tell me.
Thanks!