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'"
)
