数字类别生成onehot

对应行的列#原始标签
my_label = np.array([3,4,2,4,6,1])
#类别数量
num_class = 6
#样本数量
num = my_label.shape[0]
#生成onehot标签
label = np.zeros((num,num_class))
label[range(num),my_label-1] = 1
print(label)对应行的列#原始标签
my_label = np.array([3,4,2,4,6,1])
#类