MBZ*_*MBZ 3 testing tensorflow tensorflow-datasets
tl;博士; 我可以限制 a 中的元素数量tf.data.Dataset吗？
A 有一个训练和评估循环来处理整个给定的dataset. 这对于测试来说并不理想，因为遍历整个数据集需要很长时间。我可以通过创建 Mock 数据集或限制元素的数量来测试此代码，dataset因此代码仅通过前 10 个数据点。我该如何做第二个？
谢谢
最简单的方法只需要元件的固定数n从一个Dataset是使用Dataset.take(n)。例如：
large_dataset = ...
small_dataset = large_dataset.take(10)
| 归档时间： |
|
| 查看次数： |
437 次 |
| 最近记录： |