cerebras.modelzoo.data_preparation.utils.get_label_id_map#

cerebras.modelzoo.data_preparation.utils.get_label_id_map(label_vocab_file)[source]#

Load the label-id mapping: Mapping between output labels and id :param str label_vocab_file: Path to the label vocab file