Python - 保存模型

Python - Saving Model

提问人:keith 提问时间:11/14/2023 最后编辑:decezekeith 更新时间:11/14/2023 访问量:35

问:

我正在尝试保存我训练的模型和权重。模型将另存为包含几个文件的文件夹。如何在另一个项目中打开模型?没有 model.keras.h5 文件,只是一个包含其他几种类型文件的文件夹:

model.save_weights('model_weights.h5')
model.save('model_keras.h5')

谢谢。

蟒蛇 Keras

评论

0赞 furas 11/14/2023
我记得应该有哪个从文件夹加载数据。model.load()
0赞 furas 11/14/2023
你有没有检查过文档: 整个模型保存和加载>有特殊选项可以写成save()save_format="h5"h5

答:

1赞 furas 11/14/2023 #1

文档:整个模型的保存和加载

Model.save(filepath, overwrite=True, save_format=None, **kwargs)

save()有特殊选项可以写成save_format="h5"h5

如果没有此参数,它将以本机格式写入。
如果你使用带扩展名的文件名也没关系
Keras.h5