有谁知道如何将Tensorflow中由数据集API(tf.data.Dataset)创建的数据集拆分为Test and Train?
最佳答案
假设您具有all_dataset
类型的tf.data.Dataset
变量:
test_dataset = all_dataset.take(1000)
train_dataset = all_dataset.skip(1000)
测试数据集现在具有前1000个元素,其余的用于训练。
关于tensorflow - 将Tensorflow数据集API创建的数据集拆分为训练和测试?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/48213766/