python - 如何在 Keras 的每个时期保存训练历史？

我不能让我的电脑整天运行，为此我需要在每个时期后保存训练历史。例如，我在一天内训练了我的模型 100 个 epoch，第二天我想再训练它 50 个 epoch。我需要为整个 150 个时代生成损失与时代和准确度与时代的关系图。我正在使用 fit_generator 方法。有没有办法在每个时期之后保存训练历史(最有可能使用 Callback )？我知道如何在培训结束后保存培训历史。我正在使用 Tensorflow 后端。

最佳答案

Keras 具有 CSVLogger 回调，它似乎完全符合您的需求；来自 documentation :

它有一个附加参数用于添加到文件中。再次，从文档中:

from keras.callbacks import CSVLogger

csv_logger = CSVLogger("model_history_log.csv", append=True)
model.fit_generator(...,callbacks=[csv_logger])

关于python - 如何在 Keras 的每个时期保存训练历史？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/50127527/