Numpy反向keras to_categorical

Question

在keras中，我曾经使用to_categorical二进制nx1向量y转换为nx2矩阵，如果y = 1，则第一列为1，第二列为y = 0。如何使用numpy撤销此操作？

Answer 1

无需进行列表理解。简单地

numpy.argmax(a, axis=1)

将在所有行的每一行中找到 argmax

Answer 2

简单。

numpy.argmax(a, axis=None, out=None)

这将返回沿轴的最大值的索引。

Answer 3

添加到MazeRunner09的答案。如果您使用过来自keras的to_categorical，则将有一个列表，并且可以对整个一键编码列表使用列表理解：

y_classes = [np.argmax(y, axis=None, out=None) for y in y_test]