2

制作这个 One Hot 编码矩阵的最佳方法是什么

array([[[1, 0, 0],
        [1, 0, 0],
        [0, 1, 0]],

       [[0, 0, 1],
        [0, 1, 0],
        [1, 0, 0]]])

作为

array([[0, 0, 1],
       [2, 1, 0]])

换句话说,如何解码 One Hot 数组?

4

1 回答 1

1

np.argmax一起使用axis=2——

a.argmax(2)

样品运行 -

In [186]: a
Out[186]: 
array([[[1, 0, 0],
        [1, 0, 0],
        [0, 1, 0]],

       [[0, 0, 1],
        [0, 1, 0],
        [1, 0, 0]]])

In [187]: a.argmax(2)
Out[187]: 
array([[0, 0, 1],
       [2, 1, 0]])
于 2017-03-25T14:53:11.027 回答