MNIST keras 模型的错误预测

Question

我是神经网络的新手，所以我尝试了我的第一个神经网络，它与 keras 学习页面上的神经网络非常接近，如下所示：

https://github.com/aakarsh1011/Neural-Network/blob/master/MNSIT%20classification.ipynb

请看一下结尾，我将随机图像红色并试图预测它作为一个袋子出来，当在epocs=5 训练时，它预测它是一个凉鞋。 我的代码或标签有问题。

更新 - 作为该领域的新手，我不知道时代的重要性，所以我问了这个问题，我担心我不会过度拟合模型或训练训练过多.但是没有明确的方法可以做到这一点，这都是尝试和错误。祝你好运！

Answer 1

首先，据我所知，您的代码是正确的。您的模型预测错误的项目可能是由于模型训练时间不够长造成的。我强烈建议您设置epochs=100，您将能够看到模型的准确性提高。通常，您应该始终尝试为您的模型提供尽可能多的 epochs 进行训练。这只需要一些时间。尝试一些不同数量的epochs，以找到不会花费太长时间但仍能得到可接受结果的那个。