def vis(doc_name, list_of_dict): from visualization import Visualization import pandas as pd triplet_list = [] for doc in list_of_dict: triplet_list.append([ doc['subject'], doc['relation'], doc['object'], doc['subj_type'], doc['obj_type'] ]) triplet_DF = pd.DataFrame( triplet_list, columns=['subject', 'relation', 'object', 'subj_type', 'obj_type']) Visualization.drawKnowledgeGraph(triplet_DF) Visualization.queryKnowledgeGraph(triplet_DF, doc_name) # Storing the data in pandas Dataframe if (DEBUG['KG2csv']): triplet_DF.to_csv("./textual_data/list_of_entities.csv") print( "[INFO] Extracted knowledge has been stored in './textual_data/list_of_entities.csv'" )