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装饰器
May 09 Python
Python编码爬坑指南(必看)
Jun 10 Python
Python实现 多进程导入CSV数据到 MySQL
Feb 26 Python
django数据库migrate失败的解决方法解析
Feb 08 Python
Python中矩阵创建和矩阵运算方法
Aug 04 Python
Python开启线程,在函数中开线程的实例
Feb 22 Python
Jupyter Notebook 实现正常显示中文和负号
Apr 24 Python
Matplotlib中%matplotlib inline如何使用
Jul 28 Python
django创建css文件夹的具体方法
Jul 31 Python
Scrapy基于scrapy_redis实现分布式爬虫部署的示例
Sep 29 Python
Python爬虫UA伪装爬取的实例讲解
Feb 19 Python
PyCharm 安装与使用配置教程(windows,mac通用)
May 12 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
与空气斗智斗勇的经典《Overlord》,传说中的“无稽之谈”
2020/04/09 日漫
谈一谈收音机的高放电路
2021/03/02 无线电
用PHP实现图象锐化代码
2007/06/14 PHP
PHP读取MySQL数据代码
2008/06/05 PHP
php更新mysql后获取影响的行数发生异常解决方法
2013/03/28 PHP
php实现cc攻击防御和防止快速刷新页面示例
2014/02/13 PHP
javascript特殊用法示例介绍
2013/11/29 Javascript
JavaScript实现两个Table固定表头根据页面大小自行调整
2014/01/03 Javascript
js给selected添加options的方法
2015/05/06 Javascript
基于Jquery代码实现支持PC端手机端幻灯片代码
2015/11/17 Javascript
js实现分页功能
2017/05/24 Javascript
网页中的图片查看器viewjs使用方法
2017/07/11 Javascript
解决基于 keep-alive 的后台多级路由缓存问题
2020/12/23 Javascript
html5以及jQuery实现本地图片上传前的预览代码实例讲解
2021/03/01 jQuery
python中对list去重的多种方法
2014/09/18 Python
Python使用py2exe打包程序介绍
2014/11/20 Python
Python 判断奇数偶数的方法
2018/12/20 Python
决策树剪枝算法的python实现方法详解
2019/09/18 Python
Pytorch 神经网络—自定义数据集上实现教程
2020/01/07 Python
Python3爬虫中识别图形验证码的实例讲解
2020/07/30 Python
在终端启动Python时报错的解决方案
2020/11/20 Python
html5手机端页面可以向右滑动导致样式受影响的问题
2018/06/20 HTML / CSS
HTML5如何使用SVG的方法示例
2019/01/11 HTML / CSS
StubHub希腊:购买体育赛事、音乐会和剧院门票
2019/08/03 全球购物
英国时尚配饰、珠宝和服装网站:KJ Beckett
2020/01/23 全球购物
文史专业毕业生自荐信
2013/11/17 职场文书
《桃花心木》教学反思
2014/02/17 职场文书
领导调研接待方案
2014/02/27 职场文书
教学质量评估实施方案
2014/03/17 职场文书
机关党员三严三实心得体会
2014/10/13 职场文书
前台接待岗位职责
2015/02/03 职场文书
乡镇团委工作总结2015
2015/05/26 职场文书
2015年学校医务室工作总结
2015/07/20 职场文书
2016年清明节寄语
2015/12/04 职场文书
2016入党培训心得体会范文
2016/01/08 职场文书
Consul在linux环境的集群部署
2022/04/08 Servers