gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-28 20:22:05 25 4
gpt4 key购买 nike

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

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

谢谢

最佳答案

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

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

关于testing - 限制 tf.data.Dataset 中的项目数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51006470/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com