如何使tf.data.Dataset只改组数据列的子集?

时间:2019-09-20 08:45:47

标签: tensorflow tensorflow-datasets

让我们看看有一个数据集:

   ds = tf.data.Dataset.from_tensor_slices((x_train,y_train ))

如果我应用ds = ds.shuffle(),则x_train和y_train都将被改组。如果我只想在不改变y_train的情况下改组x_train怎么办?

2 个答案:

答案 0 :(得分:1)

您可以使用仅返回1个输出的map函数来实现您想要的功能。然后您可以将它们压缩。

.transactionally

答案 1 :(得分:0)

由于将错误的标签映射到要素,因此不确定为什么要这样做。 但是无论如何,如果您仍然想这样做,则可以先将x_train改编,再将其输入from_tensor_slices()