tfma.model_util.get_label_key

View source on GitHub

Returns the label_key corresponding to a given output name.