NER-Evaluation
NER-Evaluation copied to clipboard
true_which_overlapped_with_pred does not get updated properly
Hi,
I think there are few scenarios that you are not updating true_which_overlapped_with_pred properly, and break your for loops.
For example in line 267 of ner_eval.py file, you are appending a "true" entity to true_which_overlapped_with_pred. and then set found_overlap = True and then break. In this case you are not appending other "true" entities that they have partial overlap. For example if annotated data has two spans of [2,4] and [5, 7] and NER model has one span of [3, 6], both "true" spans have overlap with this one "pred" span, but you are breaking the loop, before allowing appending of entity with span [5, 7] to be added to the true_which_overlapped_with_pred. I think it is right that metrics should not get updated, but true_which_overlapped_with_pred should get updated with both "true" spans.
I tried changing the code as follows and it started counting spans properly: