I am building an NER model, I must have for each tagged entity a probability to indicate the confidence prediction for each label,
I tried to just use the model output for the last classification layer, but it seems for me it’s not a probability and all values are so close (see below).
image1045×630 45.7 KB
Any additional information would be helpful!
Thank u
Zakarya