tf.data.Dataset有哪些参数
时间: 2023-06-03 19:02:05 浏览: 87
tf.data.Dataset.map与tf.data.Dataset.interleave
tf.data.Dataset有以下参数:
1. batch_size:每个batch包含的样本数量。
2. repeat:数据集重复的次数。如果为None或-1,则表示数据集将无限次重复。
3. shuffle:是否对数据集进行打乱处理,可以指定打乱时使用的随机种子。
4. prefetch:预取数量,可以在训练时提高数据读取效率。
5. num_parallel_calls:并行调用数量,可以在数据预处理时提高效率。
6. drop_remainder:最后一个batch是否丢弃,如果为True,则丢弃,否则保留不足batch_size的部分。
阅读全文