tfma.model_util.get_label_key

Returns the label_key corresponding to a given output name.