impresso-project
/

ner-newsagency-bert-fr

Token Classification

Model card Files Files and versions

emanuelaboros commited on Sep 30, 2024

Commit

84dbdc0

·

verified ·

1 Parent(s): f90ef1c

Update newsagency_ner.py

Files changed (1) hide show

newsagency_ner.py +1 -1

newsagency_ner.py CHANGED Viewed

@@ -130,7 +130,7 @@ def get_entities(tokens, tags, confidences, text):
                 entities.append(
                     {
                         "entity": original_label,
-                        "score": np.average(confidences[idx : idx + len(subtree)]),
                         "index": idx,
                         "word": original_string,
                         "start": entity_start_position,

                 entities.append(
                     {
                         "entity": original_label,
+                        "score": np.round(np.average(confidences[idx : idx + len(subtree)]) * 100.0, 2),
                         "index": idx,
                         "word": original_string,
                         "start": entity_start_position,