def rejection_resample(self, ds):
nclasses = 1000
def _classfunc(*tensors):
as_dict = self.dictify(tensors)
uids = as_dict['uid']
return tf.mod(uids, nclasses)
target_dist = tf.constant(1/nclasses, shape=(nclasses,))
return tf.contrib.data.rejection_resample(ds, _classfunc, target_dist)
dataset.py 文件源码
python
阅读 32
收藏 0
点赞 0
评论 0
评论列表
文章目录