我的数据集结构如下:[...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()
结合使用呢?
回答:
将keras.utils.to_categorical
替换为tf.one_hot
应该可以解决这个问题。