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