一个热编码变量n m模式编号

Question

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

愚蠢的问题但是，给定标签(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