in main.py [0:0]
def calculate_prf(all_gts, all_pds):
ans = calculate_prf_one_group(all_gts, all_pds)
constraint_stats = {}
for constraint in all_constraints:
cname = constraint['name']
this_gts = set([t for t in all_gts if t[-1] == cname])
this_pds = set([t for t in all_pds if t[-1] == cname])
constraint_stats[cname] = calculate_prf_one_group(this_gts, this_pds)
ans['constraint_stats'] = constraint_stats
return ans