train-scripts / calc_avg_comet.py
Ashton2000's picture
Upload folder using huggingface_hub
981b783 verified
import sys
with open(sys.argv[1], 'r') as f:
lines = [line.strip() for line in f]
scores = [float(line.split()[-1]) for line in lines]
with open(sys.argv[1], 'a') as f:
f.write(f'Average score: {sum(scores) / len(scores) * 100:.2f}\n')