Update relation extraction log messages

AuvaLab · Jul 9, 2024 · 4cee17c · 4cee17c
1 parent 2eed441
commit 4cee17c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/itext2kg/graph_integration/itext2kg.py b/itext2kg/graph_integration/itext2kg.py
@@ -35,12 +35,12 @@ def extract_entities_for_all_sections(self, sections:List[str], ent_threshold =
 
 
     def extract_relations_for_all_sections(self, sections:List[str], entities, rel_threshold = 0.8):
-        print("[INFO] Extracting Entities from the Document", 1)
+        print("[INFO] Extracting Relations from the Document", 1)
 
         global_relationships = self.irelations_extractor.extract_relations(context=sections[0], entities = entities)
 
         for i in range(1, len(sections)):
-            print("[INFO] Extracting Entities from the Document", i+1)
+            print("[INFO] Extracting Relations from the Document", i+1)
             entities = self.irelations_extractor.extract_relations(context= sections[i], entities=entities)
             processed_relationships, global_relationships_ = self.matcher.process_lists(list1 = entities, list2=global_relationships, for_entity_or_relation="relation", threshold = rel_threshold)