你能用'keras.utils.to_categorical'在张量流中映射一个数据集吗？

Question

我这里的数据集由以下结构中的数据组成 [...x, y] 我想将其转换为

[...x], categorical([y])

这是我尝试过的：

def map_sequence(sequence):
    return sequence[:-1], keras.utils.to_categorical(sequence[-1])

dataset = tf.data.Dataset.from_tensor_slices(input_sequences)
dataset = dataset.map(map_sequence)

但我收到一个错误，因为sequence 在执行映射时实际上没有任何数据。

如何同时使用to_categorical 和map()？

Answer 1

用tf.one_hot 替换keras.utils.to_categorical 应该可以解决问题。