什么是将Keras模型参数和模型架构与模型一起存储的最佳方法?

时间:2017-08-13 02:13:08

标签: machine-learning tensorflow neural-network keras

我希望将所有模型参数(优化器,学习速率,批量大小等)和模型架构(层的数量和类型)与模型一起保存,以便以后回去分析为什么有些模型效果更好。

是否有一种简单的方法可以将此元数据与权重一起存储?

1 个答案:

答案 0 :(得分:2)

来自the docs

  this.navCtrl.push(SearchPage, { employeeModel: this.employeeModel });

这包括优化器(应包括学习速率和批量大小)。除此之外,你可以使用

  • 使用受版本控制的配置脚本(我是为我的硕士论文做的,请参阅my configuration scripts
  • 存储培训脚本

如果您想要一个文件,只需使用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') 等容器文件格式。