TensorFlow入门(六、模型的保存和载入)

保存模型

使用TensorFlow的saver()类先实例化一个saver对象,然后在session中通过saver的save方法将模型保存起来。代码示例如下:

#初始化所有变量
init = tf.global_variable_initializer()#定义saver和保存路径
saver = tf.train.Saver()
saverdir = "save_path"#启动Session
with tf.Session() as sess:sess.run(init)#使用saver的save方法保存saver.save(sess,saverdir + "file_name")

        其中,filename如果不存在,程序会自动创建。

打印模型中的内容

使用inspect_checkpoint包中的print_tensors_in_checkpoint_file方法将模型中的具体内容打印出来。代码示例如下:

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
form tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_filesaverdir = "log/"
print_tensors_in_checkpoint_file(savedir + "linearmodel.cpkt",None,True)

保存模型的其他方法

使用saver()类保存模型时,可以在函数中放入参数来实现更高级的功能,如指定存储变量名字与变量的对应关系。代码示例如下:

W = tf.Variable(1.0,name = "weight")
b = tf.Variable(2.0,name = "bias")saver = tf.train.Saver({'weight':W,'bias':b})
with tf.Session() as sess:tf.global_variables_initializer().run()saver.save(sess,savedir + "linearmodel.cpkt")
print_tensors_in_checkpoint_file(savedir + "linearmodel.cpkt",None,True)

载入模型

通过调用saver的restore()函数,从指定的路径找到模型文件,并覆盖到相关参数中。代码示例如下:

#初始化所有变量
init = tf.global_variable_initializer()#定义saver和保存路径
saver = tf.train.Saver()
saverdir = "save_path"#启动Session
with tf.Session() as sess:sess.run(init)#使用saver的restore方法载入模型print("x=0.2,z=",sess.run(z,feed_dict = {X:0.2}))

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/88272.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

二叉树递归回溯

1、一般来说大部分二叉树题目不需要回溯。 2、路径问题大都回溯,回溯就在递归函数的后面。 回溯和递归是一一对应的,有一个递归,就要有一个回溯, 所以回溯要和递归永远在一起 二叉树所有路径 找树左下角的值 if(root->righ…

项目开发过程中,成员提离职,怎么办?

之前写过一篇《如何应对核心员工提离职》反响特别好,今天做个延展篇,在项目过程中,员工突然提离职,我们有什么办法让项目按时按质的上线。 项目做多了,总会碰到这种情况。这里给大家介绍一个解决项目问题的分析方法&a…

一文教你学会ArcGIS Pro地图设计与制图系列全流程(2)

ArcGIS Pro做的成果图及系列文章目录: 系列文章全集: 《一文教你学会ArcGIS Pro地图设计与制图系列全流程(1)》《一文教你学会ArcGIS Pro地图设计与制图系列全流程(2)》《一文教你学会ArcGIS Pro地图设计与…

Apache Doris 行列转换可以这样玩

行列转换在做报表分析时还是经常会遇到的,今天就说一下如何实现行列转换吧。 行列转换就是如下图所示两种展示形式的互相转换 1. 行转列 我们来看一个简单的例子,我们要把下面这个表的数据,转换成图二的样式 image-20230914151818953.png …

漫谈:C语言 C++ 函数返回值究竟是什么

函数的返回值经常很令人困惑,有些函数有返回值,有些函数没有返回值,有返回值的函数也可以没有return语句而正常运行。 比如这样的代码一般是可以编译的,执行也正常: int f(int a) {printf("%d\n",a); } int…

负载均衡器监控

什么是负载均衡器 负载均衡建立在现有网络结构之上,它提供了一种廉价有效透明的方法扩展网络设备和服务器的带宽、增加吞吐量、加强网络数据处理能力、提高网络的灵活性和可用性。其意思就是分摊到多个操作单元上进行执行,例如Web服务器、FTP服务器、企…

[React] 自定义hooks设计模式

文章目录 1.自定义hooks设计1.1 自定义hooks1.2 设计一个自定义hooks1.3 自定义hooks的驱动条件1.4 自定义hooks的通用模式1.5 自定义hooks的条件限定 1.自定义hooks设计 react-hooks是react16.8以后,react新增的钩子API,目的是增加代码的可复用性&…

小程序为什么必须使用SSL证书?

随着互联网技术的发展,越来越多的网站和应用程序开始使用SSL证书来保护用户数据的安全。微信小程序作为一款广受欢迎的应用程序,也必须使用SSL证书来确保用户数据的安全。 首先,使用SSL证书可以保护用户数据的安全。SSL证书是一种数字证书&am…

rust特性

特性,也叫特质,英文是trait。 trait是一种特殊的类型,用于抽象某些方法。trait类似于其他编程语言中的接口,但又有所不同。 trait定义了一组方法,其他类型可以各自实现这个trait的方法,从而形成多态。 一、…

玩转Mysql系列 - 第22篇:mysql索引原理详解

这是Mysql系列第22篇。 背景 使用mysql最多的就是查询,我们迫切的希望mysql能查询的更快一些,我们经常用到的查询有: 按照id查询唯一一条记录 按照某些个字段查询对应的记录 查找某个范围的所有记录(between and) …

学GoWorld,go 1.21

win11,下载go安装包运行, 环境变量GOPATHd:\go_work d:\go_work路径下执行 go install github.com/xiaonanln/goworldlatest 就自动下载了很多文件 进入D:\go_work\pkg\mod\github.com\xiaonanln\goworldv0.1.6 mod init goworld replace github.…

5+单细胞+脂质代谢+预后模型+实验

今天给同学们分享一篇5单细胞脂质代谢预后模型实验的生信文章“Single-cell transcriptome analysis reveals the metabolic changes and the prognostic value of malignant hepatocyte subpopulations and predict new therapeutic agents for hepatocellular carcinoma”&am…

uniapp ui安装 阿里图标库使用 报错 Assignment to constant variable.

安装 ui uni-app官网 (dcloud.net.cn) (一)安装 pages.js配置 安装 sassnpm i sass -D 或 yarn add sass -D 安装 sass-loader npm i sass-loader10.1.1 -D 或 yarn add sass-loader10.1.1 -D安装 uni-uinpm i dcloudio/uni-ui 或 yarn a…

基于若依ruoyi-nbcio增加flowable流程待办消息的提醒,并提供右上角的红字数字提醒(四)

更多ruoyi-nbcio功能请看演示系统 gitee源代码地址 前后端代码: https://gitee.com/nbacheng/ruoyi-nbcio 演示地址:RuoYi-Nbcio后台管理系统 上一节说到待办系统的监听器TaskCreateListener,需要在flowable全局监听配置里加入配置 1、Glo…

【开发篇】八、SpringBoot整合MongoBD

文章目录 1、整合2、简单示例3、一点思考4、MongoDB的安装5、MongoDB的CRUD语法 1、整合 导入MongoBD的起步依赖:(这个starter背后是MongoDB的驱动和其他依赖,在这儿也可以看出命名的规律,redis的就是spring-boot-starter-data-r…

(Mysql高级语句(进阶查询语句+数据库函数+连接查询))

Mysql高级语句(进阶查询语句MySQL数据库函数连接查询) 一、mysql查询语句1.1、 select ----显示表格中一个或数个字段的所有数据记录1.2、 distinct ----不显示重复的数据记录1.3、where ----有条件查询1.4、 and or ----且 或1.5 、in----显示已知的值的…

oracle 比 mysql 查询快的原因_Oracle 查询速度慢的原因总结

目录 1. oracle 比 mysql 查询快的原因_Oracle 查询速度慢的原因总结 1. oracle 比 mysql 查询快的原因_Oracle 查询速度慢的原因总结 查询速度慢的原因很多, 常见如下几种: 没有索引或者没有用到索引(这是查询慢最常见的问题, 是程序设计的缺陷)I/O 吞吐量小, 形成了瓶颈效应…

算法竞赛备赛之动态规划训练提升,DP基础掌握

1.背包问题 1.1.01背包问题 01背包问题是在M件物品中选择若干件放在空间为W的背包中,每件物品的体积为W1,W2至Wn,价值为P1,P2至Pn,01背包的约束条件是给定几种物品,每种物品有且只有一个,并且…

ansible的个人笔记使用记录

1.shell模块使用,shell模块------执行命令,支持特殊符 ansible all -m shell -a yum -y install nginx ansible all -m shell -a systemctl restart nginx ansible all -m shell -a systemctl stop nginx && yum -y remove nginx2. file模块…

【JS | MD5】使用crypto模块实现md5加密功能(解决中文加密前后端不一致的问题)

正常情况下使用md5加密 var crypto require(crypto);var md5Sign function (data) {var md5 crypto.createHash(md5).update(data).digest(hex);return md5; } 实际开发中经常需要前端nodejs调用后端java接口,使用上述方法会出现中文加密结果不同的情况&#x…