Spaces:
Runtime error
Runtime error
File size: 312 Bytes
fadb92b | 1 2 3 4 5 6 7 | def compute_f1(quant_result_dict, metric_category):
for metric in metric_category:
prec = quant_result_dict[metric + "_prec"]
rec = quant_result_dict[metric + "_rec"]
f1 = 2 * prec * rec / (prec + rec + 1e-5)
quant_result_dict[metric + "_f1"] = f1
return quant_result_dict |