您的位置:首页 > 大数据 > 人工智能

tf.train.batch()

2017-10-25 15:02 162 查看
tf.train.batch([example, label],batch_size=batch_size, capacity=capacity)

 

[example, label]表示样本和样本标签,这个可以是一个样本和一个样本标签,batch_size是返回的一个batch样本集的样本个数。capacity是队列中的容量。这主要是按顺序组合成一个batch

 

tf.train.shuffle_batch([example, label],batch_size=batch_size, capacity=capacity, min_after_dequeue)

 

这里面的参数和上面的一样的意思。不一样的是这个参数min_after_dequeue,一定要保证这参数大于capacity参数的值,否则会出错。这个代表队列中的元素大于它的时候就输出乱的顺序的batch。也就是说这个函数的输出结果是一个乱序的样本排列的batch,不是按照顺序排列的。

上面的函数返回值都是一个batch的样本和样本标签,只是一个是按照顺序,另外一个是随机的

 
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: