phiph's picture
Upload folder using huggingface_hub
7382c66 verified
# Authors: Jing He, Haodong Li
# Last modified: 2025-10-10
import os
def init_per_sample_csv(output_dir, alignment, metric_funcs):
per_sample_csv = os.path.join(output_dir, f"per_sample_{alignment}.csv")
with open(per_sample_csv, "w+") as f:
f.write("filename,")
f.write(",".join([m.__name__ for m in metric_funcs]))
f.write("\n")
return per_sample_csv
def write_per_sample_csv(per_sample_csv, pred_name, sample_metric):
with open(per_sample_csv, "a+") as f:
f.write(pred_name + ",")
f.write(",".join(sample_metric))
f.write("\n")
def write_metrics_txt(eval_dir, alignment, eval_text):
metrics_filename = f"eval_metrics_{alignment}.txt"
_save_to = os.path.join(eval_dir, metrics_filename)
with open(_save_to, "w+") as f:
f.write(eval_text)