有谁知道如何将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/

10-12 22:21
查看更多