在 Tensorflow 中批量训练模型的快速示例？

Question

我在谷歌上搜索了很多，我找不到一个批量训练模型的例子。 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.....})

类似的东西会起作用。