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 相关文章推荐
使用graphics.py实现2048小游戏
Mar 10 Python
python如何实现excel数据添加到mongodb
Jul 30 Python
python的一些加密方法及python 加密模块
Jul 11 Python
pip install python 快速安装模块的教程图解
Oct 08 Python
python 解决flask uwsgi 获取不到全局变量的问题
Dec 22 Python
Django密码存储策略分析
Jan 09 Python
python各层级目录下import方法代码实例
Jan 20 Python
运行tensorflow python程序,限制对GPU和CPU的占用操作
Feb 06 Python
Python csv文件记录流程代码解析
Jul 16 Python
Python私有属性私有方法应用实例解析
Sep 15 Python
python和node.js生成当前时间戳的示例
Sep 29 Python
python中HTMLParser模块知识点总结
Jan 25 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函数篇之掌握ord()与chr()函数应用
2011/12/05 PHP
php使用json_encode对变量json编码
2014/04/07 PHP
php微信公众号开发之微信企业付款给个人
2018/10/04 PHP
Laravel配置全局公共函数的方法步骤
2019/05/09 PHP
浅析PHP反序列化中过滤函数使用不当导致的对象注入问题
2020/02/15 PHP
Jquery操作Select 简单方便 一个js插件搞定
2009/11/12 Javascript
JavaScript函数定义的常见注意事项小结
2014/09/16 Javascript
Jquery中CSS选择器用法分析
2015/02/10 Javascript
jQuery实现鼠标经过图片变亮其他变暗效果
2015/05/08 Javascript
jQuery使用$.ajax进行即时验证实例详解
2015/12/11 Javascript
JavaScript ES6的新特性使用新方法定义Class
2016/06/28 Javascript
vue-ajax小封装实例
2017/09/18 Javascript
webpack中使用iconfont字体图标的方法
2018/02/22 Javascript
微信小程序开发之改变data中数组或对象的某一属性值
2018/07/05 Javascript
Vuerouter的beforeEach与afterEach钩子函数的区别
2018/12/26 Javascript
vue.js中使用echarts实现数据动态刷新功能
2019/04/16 Javascript
压缩Vue.js打包后的体积方法总结(Vue.js打包后体积过大问题)
2020/02/03 Javascript
JS面试题中深拷贝的实现讲解
2020/05/07 Javascript
PyTorch读取Cifar数据集并显示图片的实例讲解
2018/07/27 Python
对python创建及引用动态变量名的示例讲解
2018/11/10 Python
python中的TCP(传输控制协议)用法实例分析
2019/11/15 Python
python paramiko远程服务器终端操作过程解析
2019/12/14 Python
python pycharm最新版本激活码(永久有效)附python安装教程
2020/09/18 Python
python入门之基础语法学习笔记
2020/02/08 Python
python实现ftp文件传输功能
2020/03/20 Python
python实现音乐播放和下载小程序功能
2020/04/26 Python
CSS去掉A标签(链接)虚线框的方法
2014/04/01 HTML / CSS
CSS3中的Transition过度与Animation动画属性使用要点
2016/05/20 HTML / CSS
Desigual德国官网:在线购买原创服装
2018/03/27 全球购物
viagogo波兰票务平台:演唱会、体育比赛、戏剧门票
2018/04/23 全球购物
法国亚马逊官方网站:Amazon.fr
2020/12/19 全球购物
视图的作用
2014/12/19 面试题
运动会闭幕式解说词
2014/02/21 职场文书
社区娱乐活动方案
2014/08/21 职场文书
《七律·长征》教学反思
2016/02/16 职场文书
导游词之五台山
2019/10/11 职场文书