gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-30 09:34:54 25 4
gpt4 key购买 nike

我想将所有模型参数(优化器、学习率、批量大小等)和模型架构(层数和类型)与模型一起保存,以便稍后回过头来分析为什么某些模型效果更好.

是否有一种简单的方法来存储此元数据以及权重?

最佳答案

来自the docs :

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')

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

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

如果您想要一个文件,只需使用容器文件格式,例如 .tar

关于machine-learning - 将 Keras 模型参数和模型架构与模型一起存储的最佳方法是什么?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45656479/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com