代码之家 › 专栏 › 技术社区 › MBZ

限制tf.data.dataset中的项数

tensorflow-datasets tensorflow testing

MBZ · 技术社区 · 6 年前

TL;博士; 我可以限制 tf.data.Dataset 是吗?

a有一个训练和评估循环,它处理整个给定的 dataset 是的。这对于测试来说并不理想,因为要花很长时间才能遍历整个数据集。我可以通过创建模拟数据集或限制 数据集 所以代码只会遍历前10个数据点。我怎么做第二个呢?

谢谢

1 回复 | 直到 6 年前

mrry 6 年前

只取固定数量元素的最简单方法 n 从一个 Dataset 是用来 Dataset.take(n) .例如:

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

推荐文章

nevzatseferoglu · go断言实用程序函数的行为类似于非阻塞操作

2 年前

mg610 · 如何开始C++单元测试

2 年前

vidhu · 无URL的自动化测试

2 年前

DucanoidYoung · 如何在不并行运行karate-junit5的情况下切换报告备份

2 年前

Aessandro · js开关站单元测试[关闭]

6 年前

AntoineLB · 断言后期工作Django

6 年前

ravikant · Selenium脚本不工作异常

6 年前

JackMahoney · 如何在SpringBootTest集成测试期间持久化实体

6 年前

splintor Vladimir Nabokov · 如何在angular中的单元测试中测试文档点击

6 年前

SeekanDestroy · Selenium:如何将id或类与特定div关联?

6 年前