keras - 将epoch trace保存到文件

时间:2018-05-01 20:24:39

标签: keras trace epoch

我在keras上运行神经网络。我想在运行神经网络之后收回所有的纪元痕迹以进行分析。有没有一种简单的方法将下面的信息保存到txt文件?

Epoch 16/75
183/183 [==============================] - 587s 3s/step - loss: 0.0079 - mean_absolute_error: 0.0585
- categorical_accuracy: 0.9836 - val_loss: 0.0797 - val_mean_absolute_error: 0.2410 - val_categorical
_accuracy: 0.7901

Epoch 17/75
183/183 [==============================] - 587s 3s/step - loss: 0.0080 - mean_absolute_error: 0.0606
- categorical_accuracy: 0.9781 - val_loss: 0.0821 - val_mean_absolute_error: 0.2395 - val_categorical
_accuracy: 0.8148

1 个答案:

答案 0 :(得分:1)

我猜您正在寻找CSVLogger回调。

https://keras.io/callbacks/#csvlogger

csv_logger = CSVLogger('training.log')
model.fit(X_train, Y_train, callbacks=[csv_logger])