emanuelaboros
commited on
Update newsagency_ner.py
Browse files- newsagency_ner.py +1 -1
newsagency_ner.py
CHANGED
@@ -130,7 +130,7 @@ def get_entities(tokens, tags, confidences, text):
|
|
130 |
entities.append(
|
131 |
{
|
132 |
"entity": original_label,
|
133 |
-
"score": np.average(confidences[idx : idx + len(subtree)]),
|
134 |
"index": idx,
|
135 |
"word": original_string,
|
136 |
"start": entity_start_position,
|
|
|
130 |
entities.append(
|
131 |
{
|
132 |
"entity": original_label,
|
133 |
+
"score": np.round(np.average(confidences[idx : idx + len(subtree)]) * 100.0, 2),
|
134 |
"index": idx,
|
135 |
"word": original_string,
|
136 |
"start": entity_start_position,
|