繁体   English   中英

加载时期日志 Keras model

[英]Load epochs logs Keras model

我加载了 Keras model 我已经训练了 150 个 epoch

tbCallBack = tensorflow.keras.callbacks.TensorBoard(log_dir='./Graph', histogram_freq=0, write_graph=True, write_images=True)


my_model.fit(X_train, X_train,
                     epochs=200, 
                     batch_size=100,
                     shuffle=True,
                     validation_data = (X_test, X_test),
                     callbacks=[tbCallBack]
               )

# Save the model
my_model.save('my_model.hdf5')

然后,我将加载 Keras model

my_model = load_model("my_model.hdf5")

有没有办法加载所有的时代日志(损失,准确性..)?

您可以使用名为CSVLogger的 keras 回调。

根据文档,它将每个时期的结果流式传输到 csv 文件中。

这是它的文档中的代码。

from keras.callbacks import CSVLogger

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

然后,您可以根据需要将其作为普通的 CSV 文件进行操作。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM