Commit ·
0968912
1
Parent(s): f7dfd44
add comp back
Browse files- generic_ner.py +2 -2
generic_ner.py
CHANGED
|
@@ -353,7 +353,7 @@ def postprocess_entities(entities):
|
|
| 353 |
filtered_entities = repair_names_in_entities(filtered_entities)
|
| 354 |
|
| 355 |
# Step 3: Remove entities that are not useful for NEL
|
| 356 |
-
filtered_entities = clean_coarse_entities(filtered_entities)
|
| 357 |
|
| 358 |
# filtered_entities = remove_blacklisted_entities(filtered_entities)
|
| 359 |
|
|
@@ -612,7 +612,7 @@ class MultitaskTokenClassificationPipeline(Pipeline):
|
|
| 612 |
# print("After remove_included_entities:")
|
| 613 |
all_entities = remove_included_entities(all_entities)
|
| 614 |
all_entities = remove_trailing_stopwords(all_entities)
|
| 615 |
-
|
| 616 |
|
| 617 |
# print("After attach_comp_to_closest:")
|
| 618 |
# pprint(all_entities)
|
|
|
|
| 353 |
filtered_entities = repair_names_in_entities(filtered_entities)
|
| 354 |
|
| 355 |
# Step 3: Remove entities that are not useful for NEL
|
| 356 |
+
# filtered_entities = clean_coarse_entities(filtered_entities)
|
| 357 |
|
| 358 |
# filtered_entities = remove_blacklisted_entities(filtered_entities)
|
| 359 |
|
|
|
|
| 612 |
# print("After remove_included_entities:")
|
| 613 |
all_entities = remove_included_entities(all_entities)
|
| 614 |
all_entities = remove_trailing_stopwords(all_entities)
|
| 615 |
+
all_entities = postprocess_entities(all_entities)
|
| 616 |
|
| 617 |
# print("After attach_comp_to_closest:")
|
| 618 |
# pprint(all_entities)
|