array list 如何进行one-hot编码？

Question

多个样本 array list 怎么进行one-hot编码呢？

进行手动编码时报错

Answer 1

import numpy as np

labels = np.array([[1,2,3,4], [2,3,4,5], [0,4,6,2], [1,6,8,9]])
res = np.eye(10)[labels.flatten()]
print("labels转成one-hot形式的结果：\n", res, "\n")
print("labels转化成one-hot后的大小：", res.shape)