有一个简单的解决方案来解决您的问题。每次使用任何一种fit
方法-结果都会返回称为 的特殊回调。它具有一个字段history
,该字段是每个时期后注册的所有度量的字典。因此,可以在每个时期之后轻松获得损失函数值的列表:
history_callback = model.fit(params...)
loss_history = history_callback.history["loss"]
将此类列表保存到文件很容易(例如,通过将其转换为numpy
数组并使用savetxt
方法)。
尝试:
import numpy
numpy_loss_history = numpy.array(loss_history)
numpy.savetxt("loss_history.txt", numpy_loss_history, delimiter=",")