如何将tf.data.dataset类型的数据集拆分为训练数据集和验证数据集?

时间:2019-12-26 11:08:50

标签: tensorflow deep-learning neural-network tensorflow-datasets

我有一个名为combined_training_ds的{​​{1}}类型的数据集。 我正在尝试将tf.data.dataset分为combined_training_ds(共196个元素)和validation_ds(共784个元素)

train_ds

当我将此数据集传递到模型中进行训练时,出现以下错误:

  

ValueError:未知TensorShape上未定义as_list()。

在处理上述异常期间,发生了另一个异常:

发生错误

validation_ds = combined_training_ds.take(196).batch(10)
train_ds = combined_training_ds.skip(196).shuffle(784).batch(10)

0 个答案:

没有答案