degbu

generic_ner.py

@@ -403,30 +403,30 @@ class MultitaskTokenClassificationPipeline(Pipeline):
         # add titles to comp entities
         from pprint import pprint
 
-        print("Before:")
-        pprint(entities)
+        # print("Before:")
+        # pprint(entities)
 
         all_entities = []
         for key in entities:
             if key not in ["NE-COARSE-LIT"]:
                 all_entities.extend(entities[key])
 
-        print("After remove_included_entities:")
+        # print("After remove_included_entities:")
         all_entities = remove_included_entities(all_entities)
-        pprint(all_entities)
+        # pprint(all_entities)
 
-        print("After postprocess_entities")
+        # print("After postprocess_entities")
         all_entities = postprocess_entities(
             all_entities,
         )
-        pprint(all_entities)
+        # pprint(all_entities)
 
         all_entities = remove_trailing_stopwords(all_entities)
-        print("After remove_trailing_stopwords:")
-        pprint(all_entities)
+        # print("After remove_trailing_stopwords:")
+        # pprint(all_entities)
         # Attach "comp.function" entities to the closest non-"comp.function" entity
         all_entities = attach_comp_to_closest(all_entities)
-        print("After attach_comp_to_closest:")
-        pprint(all_entities)
-        print("\n")
+        # print("After attach_comp_to_closest:")
+        # pprint(all_entities)
+        # print("\n")
         return all_entities