impresso-project
/

ner-stacked-bert-multilingual-v1.1.0

Token Classification

Model card Files Files and versions

emanuelaboros commited on Oct 24, 2025

Commit

a76ce02

·

1 Parent(s): eb6e4b3

3 filters

Files changed (1) hide show

generic_ner.py +6 -3

generic_ner.py CHANGED Viewed

@@ -462,11 +462,14 @@ def postprocess_entities(entities):
     # Step 2: Attach "comp.function" entities to the closest other entities
     from pprint import pprint
-    pprint("Before attach_comp_to_closest:", filtered_entities, "\n")
     filtered_entities = attach_comp_to_closest(filtered_entities)
-    pprint("After attach_comp_to_closest:", filtered_entities, "\n")
     filtered_entities = repair_names_in_entities(filtered_entities)
-    pprint("After repair_names_in_entities:", filtered_entities, "\n")
     # Step 3: Remove entities that are not useful for NEL
     # filtered_entities = clean_coarse_entities(filtered_entities)

     # Step 2: Attach "comp.function" entities to the closest other entities
     from pprint import pprint
+    print("Before attach_comp_to_closest:")
+    pprint(filtered_entities)
     filtered_entities = attach_comp_to_closest(filtered_entities)
+    print("After attach_comp_to_closest:")
+    pprint(filtered_entities)
     filtered_entities = repair_names_in_entities(filtered_entities)
+    print("After repair_names_in_entities:")
+    pprint(filtered_entities)
     # Step 3: Remove entities that are not useful for NEL
     # filtered_entities = clean_coarse_entities(filtered_entities)