from sklearn.metrics import f1_score | |
def macro_f1_score(items): | |
unzipped_list = list(zip(*items)) | |
golds = unzipped_list[0] | |
preds = unzipped_list[1] | |
fscore = f1_score(golds, preds, average="macro") | |
return fscore | |
def micro_f1_score(items): | |
unzipped_list = list(zip(*items)) | |
golds = unzipped_list[0] | |
preds = unzipped_list[1] | |
fscore = f1_score(golds, preds, average="micro") | |
return fscore | |
def weighted_f1_score(items): | |
unzipped_list = list(zip(*items)) | |
golds = unzipped_list[0] | |
preds = unzipped_list[1] | |
fscore = f1_score(golds, preds, average="weighted") | |
return fscore | |