degbu

generic_ner.py

@@ -356,9 +356,14 @@ class MultitaskTokenClassificationPipeline(Pipeline):
             entities[task] = get_entities(words_list, preds_list, confidence_list, text)
 
         # add titles to comp entities
+        from pprint import pprint
+
+        print("Before:")
+        pprint(entities)
         postprocessed_entities = self.postprocess_entities(entities, text_sentence)
 
         # Attach "comp.function" entities to the closest non-"comp.function" entity
         postprocessed_entities = attach_comp_to_closest(postprocessed_entities)
-
+        print("After:")
+        pprint(postprocessed_entities)
         return postprocessed_entities