dataset.py 文件源码

python
阅读 32 收藏 0 点赞 0 评论 0

项目:instacart-basket-prediction 作者: colinmorris 项目源码 文件源码
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)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号