examples/research_projects/codeparrot/examples/train_complexity_predictor.py/compute_metrics
def compute_metrics(eval_pred):
    predictions, labels = eval_pred
    predictions = np.argmax(predictions, axis=1)
    return metric.compute(predictions=predictions, references=labels)