python保存模型与参数_如何导出python中的模型参数

本文介绍了如何在TensorFlow中使用tf.train.Saver()保存和读取模型,以及在Keras中使用model.save()保存和读取模型。通过示例代码详细解释了保存模型的步骤和文件结构。
摘要由CSDN通过智能技术生成

模型的保存和读取

1.tensorflow保存和读取模型:tf.train.Saver() .save()#保存模型需要用到save函数

save(

sess,

save_path,

global_step=None,

latest_filename=None,

meta_graph_suffix='meta',

write_meta_graph=True,

write_state=True

)

'''

sess: 保存模型要求必须有一个加载了计算图的会话,而且所有变量必须已被初始化。

save_path: 模型保存路径及保存名称

global_step: 如果提供的话,这个数字会添加到save_path后面,用于区分不同训练阶段的结果

'''

示例:#例子

import tensorflow as tf

import numpy as np

import os

#用numpy产生数据

x_data = np.linspace(-1,1,300)[:, np.newaxis] #转置

noise = np.random.normal(0,0.05, x_data.shape)

y_data = np.square(x_data)-0.5+noise

#输入层

x_ph = tf.placeholder(tf.float32, [None, 1])

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值