GeoLLM / Task2 /cot_intial.py
Ciallo0d00's picture
Upload folder using huggingface_hub
badcf3c verified
# import json
# # 读取JSON文件
# with open('./output/cot/cot/deepseek-ai/DeepSeek-V3_f.json', 'r', encoding='utf-8') as file:
# data = json.load(file)
# # 处理每个条目,提取Yes或No作为新的answer值
# for item in data:
# if "answer" in item:
# # 提取Yes或No
# answer = item["answer"].split('\n\n')[0] # 取第一个句号前的部分
# # answer = answer.split(',')[0]
# # answer = answer.split('.')[0]
# # answer = answer.split(',')[0]
# item["answer"] = answer
# # 保存修改后的数据
# with open('./output/cot/cot/deepseek-ai/DeepSeek-V3_f_modified.json', 'w', encoding='utf-8') as file:
# json.dump(data, file, ensure_ascii=False, indent=4)
# 导入json模块
import json
# 读取JSON文件
with open('F:/GeoLLM/Task2/output/cot1/cot/gpt-3.5-turbo_待处理1.json', 'r', encoding='utf-8') as file:
data = json.load(file)
# 处理每个条目,提取Yes或No作为新的answer值
for item in data:
if "answer" in item:
# 提取Yes或No
answer = item["answer"].split('\n\n')[0] # 取第一个句号前的部分
# answer = answer.split('\n')[0]
# answer = answer.split('.')[0]
# answer = answer.split(',')[0]
item["answer"] = answer
# 保存修改后的数据
with open('F:/GeoLLM/Task2/output/cot1/cot/gpt-3.5-turbo_model.json', 'w', encoding='utf-8') as file:
json.dump(data, file, ensure_ascii=False, indent=4)