def compute_metrics()

in containers/training/resources/train.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

def compute_metrics(eval_pred):
    logits, labels = eval_pred
    predictions = np.argmax(logits, axis=-1)
    metric = load_metric("accuracy")
    return metric.compute(predictions=predictions, references=labels)