我有一项多分类任务，并且我得到了像这样的单热类型预测
[[0, 1, 1],
[0, 1, 0],
[1, 0, 1]]
我希望将这个单热向量转换为标签，例如
[[1, 2], [1], [0, 2]]
我已经尝试过 tf.argmax，但它不起作用。那么我该如何处理呢？
使用列表理解：
oheList = [[0, 1, 1],
[0, 1, 0],
[1, 0, 1]]
[[i for i in range(len(el)) if el[i]==1] for el in oheList]
# [[1, 2], [1], [0, 2]]
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
