keras 模型、结构、权重的保存

本文介绍了如何使用Keras保存和加载模型。通过HDF5文件保存模型的结构、权重及训练配置,包括损失函数、优化器和准确率。当模型训练效果满意后,可以使用`model.save_weights('model_weights.h5')`保存权重,之后利用`model.load_weights('model_weights.h5')`加载权重。对于不同网络结构的迁移学习,可以按层名加载权重。
摘要由CSDN通过智能技术生成

      如何将训练好的网络进行保存,我们可以用pickle或cPickle来保存Keras模型,同时我们可以用下面的方法:

一、保存整个模型

model.save(filepath)将Keras模型和权重保存在一个HDF5文件中,该文件将包含:

  • 模型的结构
  • 模型的权重
  • 训练配置(损失函数,优化器,准确率等)
  • 优化器的状态,以便于从上次训练中断的地方
       前提是已经安装python的h5py包.
     
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值