in main.py [0:0]
def calculate_prf_one_group(all_gts, all_pds):
common = all_gts.intersection(all_pds)
try:
precision = len(common) / len(all_pds)
except:
precision = None
try:
recall = len(common) / len(all_gts)
except:
recall = None
try:
f1 = 2 * precision * recall / (precision + recall)
except:
f1 = None
return {
'precision': precision,
'recall': recall,
'f1': f1
}