【发布时间】:2019-03-27 11:13:13
【问题描述】:
每个班级大约有300张真实图片。具有 3 个类别的分类模型。 我使用 2000 个生成的样本和 10 个时期制作了一个模型。该模型还可以,但有许多假阴性预测。 比我想改进模型并将生成的样本数(真实图像数不变)增加到 20000。在第 6 个 epoch,准确率开始下降,最终达到 0.2
339/666 [==============>...............] - ETA: 52s - loss: 0.2762 - acc: 0.9012
340/666 [==============>...............] - ETA: 52s - loss: 0.2757 - acc: 0.9014
341/666 [==============>...............] - ETA: 52s - loss: 0.2754 - acc: 0.9015
342/666 [==============>...............] - ETA: 52s - loss: nan - acc: 0.9014
343/666 [==============>...............] - ETA: 52s - loss: nan - acc: 0.8995
344/666 [==============>...............] - ETA: 52s - loss: nan - acc: 0.8976
345/666 [==============>...............] - ETA: 51s - loss: nan - acc: 0.8955
是否过拟合? 我可以在不重新开始学习的情况下以某种方式实时阻止它吗?例如。是否有可能在每个时期后保存模型,并在这种情况下采用最佳模型。 或者至少,Keras 可以打破教学吗?
【问题讨论】:
-
没有过拟合。我认为可以使用
from keras.callbacks import ModelCheckpoint选择最佳迭代。 -
@AkshayNevrekar 回调,特别是 ModelCheckpoint 是我正在寻找的。看起来我可以很好地控制学习过程。在线课程中没有提到它,我曾经知道Keras。非常感谢。 P.S.:如果你把它写成答案,我会接受。
标签: python keras deep-learning