一维标签转onehot

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]]