one-hot-encoding - 一种热编码变量 nm 模式数

Question

我正在使用模式编号的图像标签m和n. 在对海量数据集进行全面训练之前，我还改变了 convnet 训练以采用不同的最大模式数来测试它。

愚蠢的问题，但是，给定一个标签(m,n)，我如何将其热编码为一个长度数组n*m？

谢谢。

编辑：哎呀，这实际上很简单，

modenum = %some integer
def getLabel(n,m):
    array = np.zeros((modeNum,modeNum),dtype='int8')
    array[n,m] = 1
    label = np.ndarray.flatten(array)
    return label