tensorflow输出权重值和偏差的方法
作者：林先生您好 发布时间：2021-10-01 15:56:13
标签：tensorflow,权重
使用tensorflow 训练模型时,我们可以使用 tensorflow自带的 Save模块 tf.train.Saver()来保存模型,使用方式很简单 就是在训练完模型后,调用saver.save()即可
saver = tf.train.Saver(write_version=tf.train.SaverDef.V2)
saver.save(sess, save_dir+"crfmodel.ckpt", global_step=0)
重新载入模型
saver = tf.train.Saver()
ckpt = tf.train.get_checkpoint_state(FLAGS.restore_model)
saver.restore(sess, ckpt.model_checkpoint_path)
但是这种方式保存的模型中包含特别多的信息,使保存的模型很大,其实里面有很多不是我们想要的.我们就想要里面最重要的权重信息和偏差等等数据,然后再自己写解密代码,就可以把模型应用于其他的平台,比如安卓手机.
那么我们可以使用下面的方式获取训练后的权重和偏移,
ww, bb = sess.run([self.W,self.b])
其中W,和b都是 Tensor类型的数据
with tf.name_scope('weights'):
self.W = tf.get_variable(
shape=[self.feat_size, self.nb_classes],
initializer=tf.truncated_normal_initializer(stddev=0.01),
name='weights'
# ,regularizer=tf.contrib.layers.l1_regularizer(0.1)
)
with tf.name_scope('biases'):
self.b = tf.get_variable(
shape=[self.nb_classes],
initializer=tf.truncated_normal_initializer(stddev=0.01),
name='bias'
)
tensorflow 输出权重 到csv或txt
import numpy as np
W_val, b_val = sess.run([weights_tensor, biases_tensor])
np.savetxt("W.csv", W_val, delimiter=",")
np.savetxt("b.csv", b_val, delimiter=",")
来源：http://blog.csdn.net/u013243986/article/details/77333887
0
投稿猜你喜欢
- 首先建一个access 数据库，库中有一个URLINDEX表，其中URL和Keywords字段分别添加了索引，如下：URL &nb
前言opencv中封装了一个专门用于求解cv::Mat均值的函数，即cv::mean(&cv::Mat)，该函数会得到Mat中各个通- 下面是一份在 HTML 4 Strict 和 XHTML 1.0 Strict 下必须遵守的标签嵌套规则，比如你不能在 <a>
- 这篇博客将介绍如何使用OpenCV和深度学习应用全面嵌套的边缘检测。并将对图像和视频流应用全面嵌套边缘检测，然后将结果与OpenCV的标准C
- 如下所示：import osimport reimport stringfile = open("data2.txt")
- 目录1. 配置Python环境变量2. 安装Python编辑器，并在其中配置Python3. 安装控制包uiautomator2，和其它辅助
- 下一代的 web 已经开始上路了，就在这个星期，MySpace 集成了 Google Gears，雅虎发布了新的 BrowserPlus，G
- 实这本是说明一个问题 ： 每个人在提高自己能力这件事情上， 需要持续不断地努力。以最典型的例子来看，只有通过学习，程序员才能保证不断进步。
- EF Core 是一个ORM（对象关系映射），它使 .NET 开发人员可以使用 .NET对象操作数据库，避免了像ADO.NET访问数据库的代
- DATE_FORMA T(date, format) 根据格式串format 格式化日期或日期和时间值date，返回结果串。可用DATE_F
- 万维网联盟(W3C)发布了HTML 5规格说明书的草稿 ,这是自HTML 4在十多年前发布以来的第一个主要的修订版.在这期间,随着开发者逐渐
- 当我们使用访问一个没有声明的变量时，JS会报错；而当我们给一个没有声明的变量赋值时，JS不会报错，相反它会认为我们是要隐式申明一个全局变量。
- 下面通过一段代码给大家介绍php参数过滤class mysafe{ public $logname; public $isshwomsg;
- 1、看机器配置，指三大件：cpu、内存、硬盘2、看mysql配置参数3、查系mysql行状态，可以用mysqlreport工具来查看4、查看
- 六、XML展望 任何一项新技术的产生都是有其需求背景的，XML的诞生是在HTML遇到不可克服的困难之后。近年来HTML在许多复杂的Web应用
- 简介Simon Willison来自英国，是一名经验丰富的开发人员。曾工作于Yahoo，是Web开发框架Django的创始人之一，也是Ope
- 在Thread和Process中，应当优选Process，因为Process更稳定，而且，Process可以分布到多台机器上，而Thread
- 计算机之所以能做很多自动化的任务，因为它可以自己做条件判断。比如，输入用户年龄，根据年龄打印不同的内容，在Python程序中，可以用if语句
- 下面给出ORACLE的一种实现方式，要分2步走：1. 建立 SEQUENCE CREATE [ OR REPLACE ] SEQUENCE
- 一、背景交通大数据是由交通运行管理直接产生的数据（包括各类道路交通、公共交通、对外交通的刷卡、线圈、卡口、GPS、视频、图片等数据）、交通相
