Can you please elaborate on your first statement. I tried these two variations - First one returns only recall
and second one throws and error saying tuple is not accepted
.
def compute_metrics(eval_pred):
# metric = load_metric("glue", "mrpc")
metric = load_metric("precision")
metric = load_metric("recall")
logits, labels = eval_pred
predictions = np.argmax(logits, axis=-1)
return metric.compute(predictions=predictions, references=labels)
def compute_metrics(eval_pred):
# metric = load_metric("glue", "mrpc")
metric1 = load_metric("precision")
metric2 = load_metric("recall")
logits, labels = eval_pred
predictions = np.argmax(logits, axis=-1)
return metric1.compute(predictions=predictions, references=labels), metric2.compute(predictions=predictions, references=labels)