def _metric_fn(x): labels = x["labels"] predictions = x["predictions"] return metrics.sensitivity_at_specificity(labels, predictions, 0.8)