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 相关文章推荐
python调用cmd复制文件代码分享
Dec 27 Python
使用Python内置的模块与函数进行不同进制的数的转换
Mar 12 Python
Python对文件操作知识汇总
May 15 Python
Python 专题一 函数的基础知识
Mar 16 Python
Python cookbook(字符串与文本)在字符串的开头或结尾处进行文本匹配操作
Apr 20 Python
python实现顺时针打印矩阵
Mar 02 Python
我喜欢你 抖音表白程序python版
Apr 07 Python
基于腾讯云服务器部署微信小程序后台服务(Python+Django)
May 08 Python
python3实现的zip格式压缩文件夹操作示例
Aug 17 Python
pytorch加载语音类自定义数据集的方法教程
Nov 10 Python
pytorch 实现多个Dataloader同时训练
May 29 Python
Python实现天气查询软件
Jun 07 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
封装一个PDO数据库操作类代码
2009/09/09 PHP
PHP安全性漫谈
2012/06/28 PHP
php对包含html标签的字符串进行截取的函数分享
2014/06/19 PHP
PHP中使用数组指针函数操作数组示例
2014/11/19 PHP
php图片的二进制转换实现方法
2014/12/15 PHP
php实现遍历多维数组的方法
2015/11/25 PHP
实例简介PHP的一些高级面向对象编程的特性
2015/11/27 PHP
在WordPress中使用PHP脚本来判断访客来自什么国家
2015/12/10 PHP
PHP邮箱验证示例教程
2016/06/01 PHP
PHP实现webshell扫描文件木马的方法
2017/07/31 PHP
php 替换文章中的图片路径,下载图片到本地服务器的方法
2018/02/06 PHP
将函数的实际参数转换成数组的方法
2010/01/25 Javascript
判断复选框是否被选中的两种方法
2014/06/04 Javascript
javascript实现类似于新浪微博搜索框弹出效果的方法
2015/07/27 Javascript
基于HTML+CSS,jQuery编写的简易计算器后续(添加了键盘监听)
2016/01/05 Javascript
一览画面点击复选框后获取多个id值的方法
2016/05/30 Javascript
JavaScript生成验证码并实现验证功能
2016/09/24 Javascript
浅谈vue自定义全局组件并通过全局方法 Vue.use() 使用该组件
2017/12/07 Javascript
node 解析图片二维码的内容代码实例
2019/09/11 Javascript
ES6 class类链式继承,实例化及react super(props)原理详解
2020/02/15 Javascript
[58:46]OG vs NAVI 2019国际邀请赛小组赛 BO2 第二场 8.15
2019/08/17 DOTA
二种python发送邮件实例讲解(python发邮件附件可以使用email模块实现)
2013/12/03 Python
玩转python爬虫之正则表达式
2016/02/17 Python
Python线程池模块ThreadPoolExecutor用法分析
2018/12/28 Python
PYQT5设置textEdit自动滚屏的方法
2019/06/14 Python
Pycharm快捷键配置详细整理
2020/10/13 Python
python+openCV对视频进行截取的实现
2020/11/27 Python
使用CSS3的appearance属性改变任何元素的浏览器默认风格
2012/12/24 HTML / CSS
HTML5输入框下拉菜单功能的示例代码
2020/09/08 HTML / CSS
英国厨房与餐具用品为主的设计品牌:Joseph Joseph
2018/04/26 全球购物
美国瑜伽服装和装备购物网站:Mukha Yoga
2019/02/22 全球购物
财务会计专业毕业生自荐信
2013/10/19 职场文书
陈欧的广告词
2014/03/18 职场文书
专题组织生活会方案
2014/06/15 职场文书
2014年档案管理员工作总结
2014/12/01 职场文书
珍惜时间的诗歌赏析
2019/08/23 职场文书