nataliehamptonn/Knowledge_Graph_RAG

1import json 2import csv 3 4with open('Test_Edges_Embeddings.json', 'r') as f1, open('Test_Nodes_Embeddings.json', 'r') as f2: 5    data1 = json.load(f1)6    data2 = json.load(f2)7 8output = 'comparison_retrieval2.csv'9 10with open(output, 'w', newline='') as csvfile: 11    csvwriter = csv.writer(csvfile)12 13    csvwriter.writerow([14    "Question", 15    "Correct Answer", 16    "Edges Retrieved Facts", 17    "Nodes Retrieved Facts", 18    "Edges KG RAG Answer", 19    "Nodes KG RAG Answer"20])21    22    for entry1, entry2 in zip(data1, data2):23        question = entry1.get('question')24        correct_answer = entry1.get('correct_answer')25        retrieved_facts1 = " | ".join([fact for sublist in entry1.get('Retrieved Facts', []) for fact in sublist])26        retrieved_facts2 = " | ".join([fact for sublist in entry2.get('Retrieved Facts', []) for fact in sublist])27        kg_rag_answer1 = entry1.get('KG RAG Answer')28        kg_rag_answer2 = entry2.get('KG RAG Answer')29 30        # Write the row to the CSV file31        csvwriter.writerow([32            question, 33            correct_answer, 34            retrieved_facts1, 35            retrieved_facts2, 36            kg_rag_answer1, 37            kg_rag_answer238        ])39 40print(f"CSV file '{output}' has been created.")41