问限制tf.data.Dataset中的项数
EN

Stack Overflow用户

提问于 2018-06-24 02:20:42

回答 1查看 898关注 0票数 3

tl;dr;可以限制tf.data.Dataset中元素的数量吗？

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

谢谢

回答已采纳

发布于 2018-06-24 23:24:45

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

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

票数 7

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/51006470

复制

相似问题

问限制tf.data.Dataset中的项数EN