Calculate f1-score a for Transformer model

nlp
1

def flat_accuracy(preds, labels):
pred_flat = np.argmax(preds, axis=1).flatten()
labels_flat = labels.flatten()
return np.sum(pred_flat == labels_flat) / len(labels_flat)

I have problem how change find the accuracy to f1-score

2

Do you have a solution?