tensorflow输出权重值和偏差的方法


Posted in Python onFebruary 10, 2018

使用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=",")

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
pymssql ntext字段调用问题解决方法
Dec 17 Python
Python中tell()方法的使用详解
May 24 Python
Python实现解析Bit Torrent种子文件内容的方法
Aug 29 Python
python 定时修改数据库的示例代码
Apr 08 Python
python 读取Linux服务器上的文件方法
Dec 27 Python
Python里字典的基本用法(包括嵌套字典)
Feb 27 Python
在Python中构建增广矩阵的实现方法
Jul 01 Python
python识别文字(基于tesseract)代码实例
Aug 24 Python
Python3 合并二叉树的实现
Sep 30 Python
解决Python 异常TypeError: cannot concatenate 'str' and 'int' objects
Apr 08 Python
python os模块常用的29种方法使用详解
Jun 02 Python
python 如何对logging日志封装
Dec 02 Python
详解tensorflow实现迁移学习实例
Feb 10 #Python
Python学习之Django的管理界面代码示例
Feb 10 #Python
Tensorflow 自带可视化Tensorboard使用方法(附项目代码)
Feb 10 #Python
tensorflow训练中出现nan问题的解决
Feb 10 #Python
用Eclipse写python程序
Feb 10 #Python
tensorflow建立一个简单的神经网络的方法
Feb 10 #Python
python取代netcat过程分析
Feb 10 #Python
You might like
php+mysqli使用预处理技术进行数据库查询的方法
2015/01/28 PHP
golang、python、php、c++、c、java、Nodejs性能对比
2017/03/12 NodeJs
PHP大文件分割上传 PHP分片上传
2017/08/28 PHP
jQuery使用手册之三 CSS操作
2007/03/24 Javascript
javascript instanceof 与typeof使用说明
2010/01/11 Javascript
js 文件引入实现代码
2010/04/23 Javascript
js setTimeout 常见问题小结
2013/08/13 Javascript
jquery mobile动态添加元素之后不能正确渲染解决方法说明
2014/03/05 Javascript
js使用for循环查询数组中是否存在某个值
2014/08/12 Javascript
node.js开机自启动脚本文件
2014/12/24 Javascript
jQueryUI中的datepicker使用方法详解
2016/05/25 Javascript
NodeJs使用Mysql模块实现事务处理实例
2017/05/31 NodeJs
利用vueJs实现图片轮播实例代码
2017/06/03 Javascript
使用 Node.js 模拟滑动拼图验证码操作的示例代码
2017/11/02 Javascript
mpvue跳转页面及注意事项
2018/08/03 Javascript
浅谈Vue的响应式原理
2019/05/30 Javascript
django简单的前后端分离的数据传输实例 axios
2020/05/18 Javascript
一起深入理解js中的事件对象
2021/02/06 Javascript
Python ljust rjust center输出
2008/09/06 Python
python开发之基于thread线程搜索本地文件的方法
2015/11/11 Python
Linux下多个Python版本安装教程
2018/08/15 Python
python单例模式获取IP代理的方法详解
2018/09/13 Python
Django unittest 设置跳过某些case的方法
2018/12/26 Python
钉钉群自定义机器人消息Python封装的实例
2019/02/20 Python
Python 限定函数参数的类型及默认值方式
2019/12/24 Python
Django crontab定时任务模块操作方法解析
2020/09/10 Python
html5 Canvas画图教程(7)—canvas里画曲线之quadraticCurveTo方法
2013/01/09 HTML / CSS
英国女性时尚鞋类的潮流制造者:Koi Footwear
2018/10/19 全球购物
美国沃尔玛网上超市:Walmart
2020/08/14 全球购物
解决python 输出到csv 出现多空行的情况
2021/03/24 Python
管理学专业个人求职信范文
2013/12/13 职场文书
办公室文员自荐书
2014/02/03 职场文书
《云雀的心愿》教学反思
2014/02/25 职场文书
文明生主要事迹
2014/05/25 职场文书
社区清明节活动总结
2014/07/04 职场文书
Python绘制散乱的点构成的图的方法
2022/04/21 Python