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