1. repeat()
数据集重复训练次数
2. filter
def filter(self, predicate):数据集通过predicate判定进行筛选。
'''
>>> dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
>>> dataset = dataset.filter(lambda x: x < 3)
>>> list(dataset.as_numpy_iterator())
[1, 2]
'''
Tensorflow tf.data.Dataset下的各种属性
猜你喜欢
转载自blog.csdn.net/qq_40041064/article/details/104971698
今日推荐
周排行