Python/Keras - 访问 ModelCheckpoint 回调

Python/Keras - accessing ModelCheckpoint callback

我正在使用 Keras 来预测时间序列。作为标准,我使用 20 个纪元。我想知道我的神经网络对 20 个时期中的每一个时期都预测了什么。

通过使用 model.predict 我得到了最后的预测。但是我想要所有的预测,或者至少是最后 10 个(具有可接受的错误级别)。

为了访问它,我正在尝试使用 Keras 的 ModelCheckpoint 函数,但是之后我无法访问它。我正在使用以下代码:

model=Sequential()

model.add(GRU(input_dim=col,init='uniform',output_dim=20))
model.add(Dense(10))
model.add(Dense(5))
model.add(Activation("softmax"))
model.add(Dense(1))

model.compile(loss="mae", optimizer="RMSprop")

checkpoint=ModelCheckpoint(filepath='/Users/Alex/checkpoint.hdf5')

model.fit(X=predictor_train, y=target_train, nb_epoch=20, batch_size=batch,validation_split=0.1) #best validation split at 0.1
model.evaluate(X=predictor_train, y=target_train,batch_size=batch,show_accuracy=True)

print checkpoint

客观地说,我的问题是:

非常感谢您的帮助:)

这段代码有两个问题:

  • 您没有将回调传递给模型的拟合方法。这是通过关键字参数 "callbacks".
  • 完成的
  • 文件路径应包含占位符(如“{epoch:02d}-{val_loss:.2f}”,Keras 将其与 str.format 一起使用,以便将每个纪元保存到不同的文件。

所以正确的版本应该是这样的:

checkpoint = ModelCheckpoint(filepath='/Users/Alex/checkpoint-{epoch:02d}-{val_loss:.2f}.hdf5')

model.fit(X=predictor_train, y=target_train, nb_epoch=20,
         batch_size=batch,validation_split=0.1, callbacks=[checkpoint])

您还可以在分配给该关键字的列表中添加其他类型的回调。

遗憾的是,回调对象不存储历史信息,因此无法从中恢复。