impresso-project
/

ner-stacked-bert-multilingual

Token Classification

Model card Files Files and versions

emanuelaboros commited on Oct 18, 2024

Commit

88de6a3

·

1 Parent(s): f9d6dfb

degbu

Files changed (1) hide show

generic_ner.py +1 -1

generic_ner.py CHANGED Viewed

@@ -126,7 +126,7 @@ def get_entities(tokens, tags, confidences, text):
                             np.average(confidences[idx : idx + len(subtree)]) * 100, 2
                         ),
                         "index": (idx, idx + len(subtree)),
-                        "word": text[
                             entity_start_position:entity_end_position
                         ],  # original_string,
                         "start": entity_start_position,

                             np.average(confidences[idx : idx + len(subtree)]) * 100, 2
                         ),
                         "index": (idx, idx + len(subtree)),
+                        "text": text[
                             entity_start_position:entity_end_position
                         ],  # original_string,
                         "start": entity_start_position,