tomaarsen / SpanMarkerNER

SpanMarker for Named Entity Recognition
https://tomaarsen.github.io/SpanMarkerNER/
Apache License 2.0
401 stars 28 forks source link

Inference threshold #66

Open Definelymes opened 2 months ago

Definelymes commented 2 months ago

Is there any way to pass a specific threshold when doing inference (predict)? I was checking the predictions that I got and all of them seems to have scores above 0.5, and I was wondering if I can get less scores as well in the model predict output.