Commit
·
a5ad8a0
1
Parent(s):
7795f59
add comp back
Browse files- generic_ner.py +4 -2
generic_ner.py
CHANGED
|
@@ -450,9 +450,11 @@ def postprocess_entities(entities):
|
|
| 450 |
|
| 451 |
# Step 2: Attach "comp.function" entities to the closest other entities
|
| 452 |
filtered_entities = attach_comp_to_closest(filtered_entities)
|
| 453 |
-
|
|
|
|
| 454 |
filtered_entities = repair_names_in_entities(filtered_entities)
|
| 455 |
-
|
|
|
|
| 456 |
|
| 457 |
# Step 3: Remove entities that are not useful for NEL
|
| 458 |
# filtered_entities = clean_coarse_entities(filtered_entities)
|
|
|
|
| 450 |
|
| 451 |
# Step 2: Attach "comp.function" entities to the closest other entities
|
| 452 |
filtered_entities = attach_comp_to_closest(filtered_entities)
|
| 453 |
+
if DEBUG:
|
| 454 |
+
print("After attach_comp_to_closest:", filtered_entities, "\n")
|
| 455 |
filtered_entities = repair_names_in_entities(filtered_entities)
|
| 456 |
+
if DEBUG:
|
| 457 |
+
print("After repair_names_in_entities:", filtered_entities, "\n")
|
| 458 |
|
| 459 |
# Step 3: Remove entities that are not useful for NEL
|
| 460 |
# filtered_entities = clean_coarse_entities(filtered_entities)
|