def label_num_list(train_path): train=pd.read_csv(train_path) n_classes = len(train) leaves_labels = sorted(list(set(train['label']))) class_to_num = dict(zip(leaves_labels, range(n_classes))) train_label=train['label'] num_train_label=[] for i in range(len(train)): train_label_i = train_label[i] num_label=class_to_num[train_label_i] num_train_label.append(num_label) print(num_train_label)
04-20
2502
09-17