为什么模型只在MNIST数据集的60000张图片中训练了1875张训练集图片? [duplicate]

我正在尝试创建一个简单的CNN来对MNIST数据集中的图像进行分类。模型达到了可接受的准确率,但我注意到模型在每个周期中只训练了1875张图片。这可能是什么原因造成的?如何解决这个问题?

model=models.Sequential()model.add(layers.Conv2D(filters=32,kernel_size=(3,3),activation='relu',input_shape=(28,28,1)))model.add(layers.MaxPooling2D((2,2)))model.add(layers.Conv2D(filters=64,kernel_size=(3,3),activation='relu'))model.add(layers.MaxPooling2D((2,2)))model.add(layers.Conv2D(filters=64,kernel_size=(3,3),activation='relu'))model.add(layers.Flatten())model.add(layers.Dense(64,activation='relu'))model.add(layers.Dense(10,activation='softmax'))Model: "sequential"_________________________________________________________________Layer (type)                 Output Shape              Param #   =================================================================conv2d (Conv2D)              (None, 26, 26, 32)        320       _________________________________________________________________max_pooling2d (MaxPooling2D) (None, 13, 13, 32)        0         _________________________________________________________________conv2d_1 (Conv2D)            (None, 11, 11, 64)        18496     _________________________________________________________________max_pooling2d_1 (MaxPooling2 (None, 5, 5, 64)          0         _________________________________________________________________conv2d_2 (Conv2D)            (None, 3, 3, 64)          36928     _________________________________________________________________flatten (Flatten)            (None, 576)               0         _________________________________________________________________dense (Dense)                (None, 64)                36928     _________________________________________________________________dense_1 (Dense)              (None, 10)                650  =================================================================Total params: 93,322Trainable params: 93,322Non-trainable params: 0

model.compile(optimizer=’adam’,loss=’sparse_categorical_crossentropy’,metrics=[‘accuracy’])

model.fit(train_images,train_labels,epochs=5)

Epoch 1/51875/1875 [==============================] - 55s 29ms/step - loss: 0.0760 - accuracy: 0.9776Epoch 2/51875/1875 [==============================] - 54s 29ms/step - loss: 0.0576 - accuracy: 0.9825Epoch 3/51875/1875 [==============================] - 55s 29ms/step - loss: 0.0454 - accuracy: 0.9862Epoch 4/51875/1875 [==============================] - 55s 29ms/step - loss: 0.0396 - accuracy: 0.9879Epoch 5/51875/1875 [==============================] - 55s 29ms/step - loss: 0.0336 - accuracy: 0.9900<tensorflow.python.keras.callbacks.History at 0x7f3e0ff43b70>

在Colab中的模型截图

训练后的模型截图


回答:

训练没有问题。模型是在每个包含32张图片的1875个批次上训练的,而不是1875张图片。

1875*32 = 60000张图片

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注