首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >块状反转keras to_categorical

块状反转keras to_categorical
EN

Stack Overflow用户
提问于 2017-11-20 02:47:36
回答 4查看 14.2K关注 0票数 14

在keras中,我使用to_categorical将二进制nx1向量y转换为nx2矩阵,其中第一列是1 if y=1,第二列是y=0。如何使用numpy撤销此操作?

EN

回答 4

Stack Overflow用户

回答已采纳

发布于 2017-11-20 04:10:04

很简单。

代码语言:javascript
复制
numpy.argmax(a, axis=None, out=None)

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

票数 15
EN

Stack Overflow用户

发布于 2017-11-24 22:43:18

添加到MazeRunner09的答案中。如果您使用的是keras中的to_categorical,那么您将拥有一个列表,并且可以在整个独热编码列表上使用列表理解:

代码语言:javascript
复制
y_classes = [np.argmax(y, axis=None, out=None) for y in y_test]
票数 9
EN

Stack Overflow用户

发布于 2019-07-11 06:03:37

不需要做列表理解。简单地

代码语言:javascript
复制
numpy.argmax(a, axis=1)

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

票数 8
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/47380663

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档