Keras笔记(三)

如何保存模型和权重

1. 既储存模型结构,又包含其权重和配置信息

使用keras.models.save(filepath)将Keras模型和权重保存在一个HDF5文件中,该文件包含:
模型的结构
模型的权重
训练配置(损失函数,优化器等)
优化器的状态,以便于从上次训练中断的地方开始
使用Keras.models.load_model(filepath)来重新实例化模型,如果文件储存了训练配置,还会同时完成模型的编译。
例子:

from keras.models import load_model
model.save("my_model.h5")
model=load_model('my_model.h5")
from keras import models
model_new=models.load_model("test_model.h5")
print(model_new.summary())
输出:
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
==============================&#
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值