作者热门文章
- mongodb - 在 MongoDB mapreduce 中,如何展平值对象?
- javascript - 对象传播与 Object.assign
- html - 输入类型 ="submit"Vs 按钮标签它们可以互换吗?
- sql - 使用 MongoDB 而不是 MS SQL Server 的优缺点
我有一个形状为 (100,10) 的 numpy 数组数据集。每一行都是一个单热编码。我想将其转换为形状为 (100,) 的 nd 数组,以便将每个向量行转换为一个整数,该整数表示非零索引的索引。有没有使用 numpy 或 tensorflow 的快速方法?
最佳答案
您可以使用 numpy.argmax或 tf.argmax .示例:
import numpy as np
a = np.array([[0,1,0,0],[1,0,0,0],[0,0,0,1]])
print('np.argmax(a, axis=1): {0}'.format(np.argmax(a, axis=1)))
输出:
np.argmax(a, axis=1): [1 0 3]
您可能还想查看 sklearn.preprocessing.LabelBinarizer.inverse_transform
.
关于python - 如何将 one-hot 编码转换为整数?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42497340/
我是一名优秀的程序员,十分优秀!