Question

一直在寻找，但似乎无法找到任何关于如何在TensorFlow中从单热值解码或转换回单个整数的示例。

我使用tf.one_hot并且能够训练我的模型，但是在分类后如何理解标签时有点困惑。我的数据是通过我创建的TFRecords文件输入的。我想过在文件中存储文本标签但是无法让它工作。似乎TFRecords无法存储文本字符串或者我错了。

Answer 1

您可以使用tf.argmax找出矩阵中最大元素的索引。由于您的一个热矢量将是一维的，并且只有一个1和其他0 s，因此假设您正在处理单个矢量，这将起作用。

index = tf.argmax(one_hot_vector, axis=0)

对于batch_size * num_classes的更标准矩阵，请使用axis=1获取大小为batch_size * 1的结果。

Answer 2

由于单热编码通常只是一个包含batch_size行和num_classes列的矩阵，并且每行都为零，并且对应于所选类的单个非零，您可以使用tf.argmax()恢复整数标签的向量：

BATCH_SIZE = 3
NUM_CLASSES = 4
one_hot_encoded = tf.constant([[0, 1, 0, 0],
                               [1, 0, 0, 0],
                               [0, 0, 0, 1]])

# Compute the argmax across the columns.
decoded = tf.argmax(one_hot_encoded, axis=1)

# ...
print sess.run(decoded)  # ==> array([1, 0, 3])

Answer 3

data = np.array([1, 5, 3, 8])
print(data)


def encode(data):
    print('Shape of data (BEFORE encode): %s' % str(data.shape))
    encoded = to_categorical(data)
    print('Shape of data (AFTER  encode): %s\n' % str(encoded.shape))
    return encoded


encoded_data = encode(data)
print(encoded_data)

def decode(datum):
    return np.argmax(datum)

decoded_Y = []
print("****************************************")
for i in range(encoded_data.shape[0]):
    datum = encoded_data[i]
    print('index: %d' % i)
    print('encoded datum: %s' % datum)
    decoded_datum = decode(encoded_data[i])
    print('decoded datum: %s' % decoded_datum)
    decoded_Y.append(decoded_datum)


print("****************************************")

print(decoded_Y)

Answer 4

tf.argmax已贬值（因此，此页面上答案中的所有链接均为404），现在应使用tf.math.argmax

用法：

import tensorflow as tf
a = [1, 10, 26.9, 2.8, 166.32, 62.3]
b = tf.math.argmax(input = a)
c = tf.keras.backend.eval(b)
# c = 4
# here a[4] = 166.32 which is the largest element of a across axis 0

注意：您也可以使用numpy进行此操作。

如何在Tensorflow中解码单热标签？

4 个答案: