在Tensorflow中批量訓練模型的快速示例？

Question

我在Google上搜索了很多，我找不到批量培訓模型的例子。在Tensorflow數據集中構建的mnist具有mnist.train.next_batch(BATCH_SIZE)函數，我需要這樣的函數將我的數據集分成批次併爲每次迭代加載下一批。

Answer 1

您可以分割你的列車數據如下：在每次迭代

#Generate a random batch of 200 samples 
rand_index = np.random.choice(len(x_data), size=200) 
rand_x = x_data[rand_index] 
rand_y = y_data[rand_index] 

然後，您可以通過批量培養一批。

for i in range(1000): 
    for u in range(number_of_batches): 
     sess.run(train, feed_dict={x: rand_x.....}) 

類似的東西會工作。