python - keras 中每一层的错误

Question

我试图通过带有 Keras 的张量板来可视化 CNN 中每一层的错误，以查看它们如何及时地在每一层中发生变化。如何获得每一层的错误？

标签： pythontensorflowkerastensorboard

Answer 1

损失仅在输出层中定义，以衡量您的模型对数据的拟合程度。keras 提供了一个函数来跟踪训练过程中的相关变量，称为 History()。

from keras.callbacks import History

history = History()

# define and compile your model
model.fit(..., callbacks=[history])
print(history.history)

最后一个命令向您显示训练过程中的所有跟踪值。您可以通过 get() 方法访问单个变量。要获得训练损失，您可以通过以下方式访问它

history.history.get('loss')