python - 如何遍历整个迭代器

Question

所以我在训练我的模型时创建了一个迭代器

train_iterator = datagen.flow(trainX, trainY, batch_size=64)

现在我想从中获取所有值。例如： batchX, batchY = train_iterator.next()

将带来第一组值。类似地，我希望迭代器的所有值都存储在列表或 numpy 数组中。是否有任何命令

标签： pythonkerasiterator

Answer 1

for batch_x, batch_y in train_iterator:
    print (batch_x.shape, batch_y.shape)