Update generic_ner.py
Browse files- generic_ner.py +5 -1
generic_ner.py
CHANGED
|
@@ -77,7 +77,7 @@ def get_entities(tokens, tags, confidences, text):
|
|
| 77 |
|
| 78 |
entities = []
|
| 79 |
idx: int = 0
|
| 80 |
-
|
| 81 |
for subtree in ne_tree:
|
| 82 |
# skipping 'O' tags
|
| 83 |
if isinstance(subtree, Tree):
|
|
@@ -87,6 +87,10 @@ def get_entities(tokens, tags, confidences, text):
|
|
| 87 |
for indices in find_entity_indices(text, original_string):
|
| 88 |
entity_start_position = indices[0]
|
| 89 |
entity_end_position = indices[1]
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
entities.append(
|
| 91 |
{
|
| 92 |
"entity": original_label,
|
|
|
|
| 77 |
|
| 78 |
entities = []
|
| 79 |
idx: int = 0
|
| 80 |
+
already_done = []
|
| 81 |
for subtree in ne_tree:
|
| 82 |
# skipping 'O' tags
|
| 83 |
if isinstance(subtree, Tree):
|
|
|
|
| 87 |
for indices in find_entity_indices(text, original_string):
|
| 88 |
entity_start_position = indices[0]
|
| 89 |
entity_end_position = indices[1]
|
| 90 |
+
if '_'.join([original_label, original_string, entity_start_position]) in already_done:
|
| 91 |
+
continue
|
| 92 |
+
else:
|
| 93 |
+
already_done.append('_'.join([original_label, original_string, entity_start_position]))
|
| 94 |
entities.append(
|
| 95 |
{
|
| 96 |
"entity": original_label,
|