Update generic_ner.py
Browse files- generic_ner.py +1 -1
generic_ner.py
CHANGED
|
@@ -106,7 +106,7 @@ def get_entities(tokens, tags, confidences, text):
|
|
| 106 |
entities.append(
|
| 107 |
{
|
| 108 |
"entity": original_label,
|
| 109 |
-
"score": np.average(confidences[idx : idx + len(subtree)]),
|
| 110 |
"index": idx,
|
| 111 |
"word": original_string,
|
| 112 |
"start": entity_start_position,
|
|
|
|
| 106 |
entities.append(
|
| 107 |
{
|
| 108 |
"entity": original_label,
|
| 109 |
+
"score": int(np.average(confidences[idx : idx + len(subtree)])),
|
| 110 |
"index": idx,
|
| 111 |
"word": original_string,
|
| 112 |
"start": entity_start_position,
|