首页 > 解决方案 > mnist手写分类器分数过低

问题描述

我按照本教程https://medium.com/the-andela-way/deep-learning-hello-world-e1fc53ea888来试验 Keras,源代码在这里https://github.com/sirghiny/mnist 但是我得分很低,训练时间也很短,好像模型是在很少的样本上训练的。这是终端中的输出:

Epoch 1/5
300/300 [==============================] - 12s 39ms/step - loss: 2.3791 - categorical_accuracy: 0.0899 - val_loss: 2.3104 - val_categorical_accuracy: 0.0528
Epoch 2/5
300/300 [==============================] - 11s 38ms/step - loss: 2.3326 - categorical_accuracy: 0.1060 - val_loss: 2.2920 - val_categorical_accuracy: 0.0864
Epoch 3/5
300/300 [==============================] - 10s 32ms/step - loss: 2.2891 - categorical_accuracy: 0.1315 - val_loss: 2.2742 - val_categorical_accuracy: 0.1571
Epoch 4/5
300/300 [==============================] - 9s 31ms/step - loss: 2.2510 - categorical_accuracy: 0.1576 - val_loss: 2.2569 - val_categorical_accuracy: 0.2367
Epoch 5/5
300/300 [==============================] - 9s 30ms/step - loss: 2.2174 - categorical_accuracy: 0.1889 - val_loss: 2.2397 - val_categorical_accuracy: 0.3133

Evaluating the model...

1250/1250 [==============================] - 2s 2ms/step - loss: 2.2382 - categorical_accuracy: 0.3171
938/938 [==============================] - 2s 2ms/step - loss: 2.2369 - categorical_accuracy: 0.3232

请告诉我我做错了什么?

标签: pythontensorflowkeras

解决方案


您只更新了模型重量 1500 次(epochs*number_of_batch)。

正如我们在您的日志中看到的网络仍在学习的那样,您可能希望增加 epochs 或/并减少 batch_size 以执行更多权重更新。

此外,由于 TensorFlow 最近发生了很大变化,您应该找到一个像这样的最新教程。


推荐阅读