在keras中,我使用to_categorical将二进制nx1向量y转换为nx2矩阵,其中第一列是1 if y=1,第二列是y=0。如何使用numpy撤销此操作?
发布于 2017-11-20 04:10:04
很简单。
numpy.argmax(a, axis=None, out=None)这将返回沿着轴的最大值的索引。
发布于 2017-11-24 22:43:18
添加到MazeRunner09的答案中。如果您使用的是keras中的to_categorical,那么您将拥有一个列表,并且可以在整个独热编码列表上使用列表理解:
y_classes = [np.argmax(y, axis=None, out=None) for y in y_test]发布于 2019-07-11 06:03:37
不需要做列表理解。简单地
numpy.argmax(a, axis=1)将在所有行的每一行中查找argmax
https://stackoverflow.com/questions/47380663
复制相似问题