benchmark/inference.py

import os
import json

root_dir = "/home/tianqiu/learn/benchmark/result/2025-04-08_08-42-28_GSM8K"
threhold = 100
src_data=[]
for i in range(threhold):
    file_name = os.path.join(root_dir, f"result_{i}.json")
    with open(file_name, "r") as f:
        data = json.load(f)
        src_data.append(data)

for i in range(threhold, 200):
    file_name = os.path.join(root_dir, f"result_{i}.json")
    with open(file_name, "r") as f:
        data = json.load(f)
        src_data.append(data)
        
def transform():
    pass