impresso-project
/

ner-stacked-bert-multilingual

Token Classification

Model card Files Files and versions Community

emanuelaboros commited on Oct 23, 2024

Commit

66102da

·

1 Parent(s): 9c3a2cf

add comp back

Files changed (1) hide show

generic_ner.py +4 -4

generic_ner.py CHANGED Viewed

@@ -759,13 +759,13 @@ class MultitaskTokenClassificationPipeline(Pipeline):
             print(all_entities)
         # print("After remove_included_entities:")
         all_entities = remove_included_entities(all_entities)
-        pprint("After remove_included_entities:", all_entities)
         all_entities = remove_trailing_stopwords(all_entities)
-        pprint("After remove_trailing_stopwords:", all_entities)
         all_entities = postprocess_entities(all_entities)
-        pprint("After postprocess_entities:", all_entities)
         all_entities = refine_entities_with_coarse(all_entities, coarse_entities)
-        pprint("After refine_entities_with_coarse:", all_entities)
         # print("After attach_comp_to_closest:")
         # pprint(all_entities)
         # print("\n")

             print(all_entities)
         # print("After remove_included_entities:")
         all_entities = remove_included_entities(all_entities)
+        print("After remove_included_entities:", all_entities)
         all_entities = remove_trailing_stopwords(all_entities)
+        print("After remove_trailing_stopwords:", all_entities)
         all_entities = postprocess_entities(all_entities)
+        print("After postprocess_entities:", all_entities)
         all_entities = refine_entities_with_coarse(all_entities, coarse_entities)
+        print("After refine_entities_with_coarse:", all_entities)
         # print("After attach_comp_to_closest:")
         # pprint(all_entities)
         # print("\n")