时间戳

首页 > 解决方案 > 从 tf.data 中随机抽取 2000 个元素

python - 从 tf.data 中随机抽取 2000 个元素

问题描述

假设您有一个包含 50,000 个示例的数据集，并且想要统一采样 2000 个元素。

例如：

dataset = tf.data.TFRecordDataset(path_filename_records)
dataset = (dataset
           .map(parse_record, num_parallel_calls=tf.data.experimental.AUTOTUNE)
           .batch(bs)
           .prefetch(tf.data.experimental.AUTOTUNE)
          )

你如何用 tf.data 做到这一点？

标签： pythontensorflowtensorflow2.0

解决方案

推荐阅读