训练期间未加载所有训练样本

Question

我只是从NLP开始。我从tensorflow_datasets加载了“ imdb_reviews”数据集。

有25000个测试样本，但是当我运行时，我只训练了782个样本。我没有使用batch_size，而是一次加载了整个数据集

其他超参数是：

vocab_size = 10000
input_length = 120
embedding_dims = 16

谁能告诉我我做错了什么？

Answer 1

默认情况下，tf.keras.model的fit方法会将批次大小设置为32。 https://www.tensorflow.org/api_docs/python/tf/keras/Model

由于32 * 782 = 25,024，它可能只是丢弃了最后一批。