set_label_map() is used in train method?