Spaces:
Runtime error
Runtime error
File size: 1,009 Bytes
4e22254 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import pandas as pd
import json
cidar = 'output/cidar.json'
chat = 'output/chat.json'
alpagasus = 'output/alpagasus.json'
with open(cidar, mode='r', encoding='utf-8') as file:
cidar_data = json.load(file)
cidar_df = pd.DataFrame(cidar_data)
with open(chat, mode='r', encoding='utf-8') as file:
chat_data = json.load(file)
chat_df = pd.DataFrame(chat_data)
with open(alpagasus, mode='r', encoding='utf-8') as file:
alpagasus_data = json.load(file)
alpagasus_df = pd.DataFrame(alpagasus_data)
alpagasus_df = alpagasus_df[['Sentence', 'Response']].rename(columns={'Sentence': 'instruction', 'Response': 'alpagasus_output'})
cidar_df.rename(columns={'model_output': 'cidar_output'}, inplace=True)
chat_df.rename(columns={'model_output': 'chat_output'}, inplace=True)
merged_df = pd.concat([cidar_df, alpagasus_df.drop(columns=['instruction']), chat_df.drop(columns=['instruction'])], axis=1)
merged_df.to_json('output/merged.json', orient='records', lines=False, force_ascii=False)
|