1. 程式人生 > >tf.train.batch 和tf.train.shuffle_batch的用法

tf.train.batch 和tf.train.shuffle_batch的用法

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_szie, capacity= capacity, min_after_dequeue)。

這裡面的引數和上面的一樣,不一樣的是這個引數min_after_dequeue,一定要保證這個引數小於capacity引數的值, 否則會出錯。

這個代表隊列中的元素大於它的時候就輸出亂的順序的batch。

上面的函式返回值都是一個batch的樣本和樣本標籤,只是一個是按照順序,另外一個是隨機的。