tensorflow - 你能用'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()一起怎么用？

标签： tensorflowmachine-learningkerasdataset

Answer 1

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