Keras:如何记录验证损失

ua4mk5z4  于 2022-11-13  发布在  其他
关注(0)|答案(1)|浏览(161)

注意:这是一个duplicate question,但我不是在寻找答案,而是如何更好地找到自己的答案。
我如何记录模型在不同时期的损失、训练准确度、测试损失和测试准确度?我想绘制一个图表,显示每个时期的验证损失。
我知道callback对象可以在fit中调用(),或者model.history与此有关,但是检查源代码和文档字符串对我来说只是一堵代码墙。例如,Numpy通常提供一个非常小的用例作为非常简单的实现示例。但我知道,对此的答案只是一行代码,因为这实际上只是一个输入问题。

igsr9ssn

igsr9ssn1#

正如在https://keras.io/models/sequential/#fit文档中详细描述的,当你调用model.fit时,它返回一个callbacks.History对象,你可以从它那里得到损失和其他指标:

...
train_history = model.fit(X_train, Y_train,
                    batch_size=batch_size, nb_epoch=nb_epoch,
                    verbose=1, validation_data=(X_test, Y_test))
loss = train_history.history['loss']
val_loss = train_history.history['val_loss']
plt.plot(loss)
plt.plot(val_loss)
plt.legend(['loss', 'val_loss'])
plt.show()

相关问题