与模型一起存储Keras模型参数和模型体系结构的最佳方法是什么？

Question

我想保存所有的模型参数(优化器，学习率，批大小等)。以及模型体系结构(层的数量和类型)和模型，这样以后再回过头来分析为什么某些模型工作得更好。

有一个简单的方法来存储这些元数据以及权重吗？

Answer 1

来自医生们

from keras.models import load_model

model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'
del model  # deletes the existing model

# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

这包括优化器(应该包括学习速度和批处理大小)。除此之外，你可以用

• 使用版本控制下的配置脚本(关于我的硕士论文，请参阅我的配置脚本)
• 存储培训脚本

如果您想要一个文件，只需使用像.tar这样的容器文件格式。