labels = [1,0,1,1,0] new_label = []

一维变二维

for j in labels:

new_label.append([j])

print(“new_label===”, new_label)

对标签形式进行重构，转换成独热编码

def onehot(labels): n_sample = len(labels) n_class = max(labels) + 1 onehot_labels = np.zeros((n_sample, n_class)) onehot_labels[np.arange(n_sample), labels] = 1 return onehot_labels

labels = onehot(labels)

print(“new_label===”, labels)

<-------------------------------------------------------------------------> tensor一维转二维用reshape 如 [1.0,2.0]转为[ [1.0,2.0]]