impresso-project
/

ner-stacked-bert-multilingual

Token Classification

Model card Files Files and versions Community

emanuelaboros commited on Oct 22, 2024

Commit

a1fe592

·

1 Parent(s): f401b99

remove previous offsets

Files changed (1) hide show

generic_ner.py +1 -1

generic_ner.py CHANGED Viewed

@@ -125,7 +125,7 @@ def get_entities(tokens, tags, confidences, text):
                 entities.append(
                     {
                         "type": original_label,
-                        "score": round(
                             np.average(confidences[idx : idx + len(subtree)]) * 100, 2
                         ),
                         "index": (idx, idx + len(subtree)),

                 entities.append(
                     {
                         "type": original_label,
+                        "confidence_ner": round(
                             np.average(confidences[idx : idx + len(subtree)]) * 100, 2
                         ),
                         "index": (idx, idx + len(subtree)),