testing - 限制 tf.data.Dataset 中的项目数

Question

tl;博士; 我可以限制 a 中的元素数量tf.data.Dataset吗？

A 有一个训练和评估循环来处理整个给定的dataset. 这对于测试来说并不理想，因为它需要很长时间才能遍历整个数据集。我可以通过创建 Mock 数据集或限制元素的数量来测试此代码，dataset因此代码仅通过前 10 个数据点。我该怎么做第二个？

谢谢

标签： testingtensorflowtensorflow-datasets

Answer 1

n从 a中仅获取固定数量元素的最简单方法Dataset是使用Dataset.take(n). 例如：

large_dataset = ...
small_dataset = large_dataset.take(10)