注意:这是duplicate question,但我没有找到答案。相反,如何更好地找到自己的答案。
如何从模型中跨越时代记录损失,训练准确性,测试损失和测试准确度?我想绘制一个图表,显示每个时期的验证损失。
我知道可以在fit()中调用的callback对象,或者也许是model.history与它有关,但检查源代码和文档字符串对我来说只是一堵代码墙。例如,Numpy通常提供一个非常小的用例作为非常简单的实现的示例。然而我知道答案只是一个单行,因为这只是一个输入问题。
答案 0 :(得分:8)
详见文档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()