debug

generic_ner.py

@@ -367,7 +367,7 @@ class MultitaskTokenClassificationPipeline(Pipeline):
             entities[task] = get_entities(words_list, preds_list, confidence_list, text)
 
         # add titles to comp entities
-        from pprint import pprint
+        # from pprint import pprint
 
         # print("Before:")
         # pprint(entities)
@@ -381,16 +381,14 @@ class MultitaskTokenClassificationPipeline(Pipeline):
         all_entities = remove_included_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
         # print("After postprocess_entities")
         all_entities = postprocess_entities(all_entities)
 
-        print("After attach_comp_to_closest:")
-        pprint(all_entities)
+        # print("After attach_comp_to_closest:")
+        # pprint(all_entities)
         # print("\n")
         return all_entities