import sklearn.metrics import numpy as np def f1_macro(targets, predictions): targets, predictions = np.asarray(targets).astype(int), np.asarray(predictions).astype(int) return {"f1_macro": 100*sklearn.metrics.f1_score(targets, predictions, average='macro')}