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中文件操作简明介绍
Apr 13 Python
Python验证企业工商注册码
Oct 25 Python
Python对多属性的重复数据去重实例
Apr 18 Python
对python 合并 累加两个dict的实例详解
Jan 21 Python
PyQt QCombobox设置行高的方法
Jun 20 Python
django框架两个使用模板实例
Dec 11 Python
python读取ini配置文件过程示范
Dec 23 Python
在PyCharm中实现添加快捷模块
Feb 12 Python
AUC计算方法与Python实现代码
Feb 28 Python
Python ConfigParser模块的使用示例
Oct 12 Python
Django如何继承AbstractUser扩展字段
Nov 27 Python
Appium中scroll和drag_and_drop根据元素位置滑动
Feb 15 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数组查找函数in_array()、array_search()、array_key_exists()使用实例
2014/04/29 PHP
thinkphp视图模型查询提示ERR: 1146:Table 'db.pr_order_view' doesn't exist的解决方法
2014/10/30 PHP
Laravel 模型使用软删除-左连接查询-表起别名示例
2019/10/24 PHP
Jquery中Event对象属性小结
2015/02/27 Javascript
swtich/if...else的替代语句
2015/08/16 Javascript
AngularJs中route的使用方法和配置
2016/02/04 Javascript
js遍历map javaScript遍历map的简单实现
2016/08/26 Javascript
javascript iframe跨域详解
2016/10/26 Javascript
js中的DOM模拟购物车功能
2017/03/22 Javascript
jQuery插件FusionCharts绘制的2D条状图效果【附demo源码】
2017/05/13 jQuery
前端防止用户重复提交js实现代码示例
2018/09/07 Javascript
jQuery操作cookie的示例代码
2019/06/05 jQuery
python实现TCP服务器端与客户端的方法详解
2015/04/30 Python
python过滤字符串中不属于指定集合中字符的类实例
2015/06/30 Python
python使用mysql的两种使用方式
2018/03/07 Python
python3爬虫之设计签名小程序
2018/06/19 Python
详解python中自定义超时异常的几种方法
2019/07/29 Python
基于Python词云分析政府工作报告关键词
2020/06/02 Python
python进度条显示之tqmd模块
2020/08/22 Python
plt.figure()参数使用详解及运行演示
2021/01/08 Python
详解css3 Transition属性(平滑过渡菜单栏案例)
2017/09/05 HTML / CSS
阿拉伯书店:Jamalon
2019/07/24 全球购物
Timberland法国官网:购买靴子、鞋子、衣服、夹克和配饰
2019/11/30 全球购物
类成员函数的重载、覆盖和隐藏区别
2016/01/27 面试题
秋季运动会广播稿大全
2014/02/17 职场文书
学校招生宣传广告词
2014/03/19 职场文书
电脑售后服务承诺书
2014/03/27 职场文书
《去年的树》教学反思
2014/04/11 职场文书
班级文化建设标语
2014/06/23 职场文书
模具专业求职信
2014/06/26 职场文书
个人年终总结结尾
2015/03/06 职场文书
中英文求职信范文
2015/03/19 职场文书
golang 在windows中设置环境变量的操作
2021/04/29 Golang
MySQL Innodb索引机制详细介绍
2021/11/23 MySQL
Python实现提取PDF简历信息并存入Excel
2022/04/02 Python
Python实现日志实时监测的示例详解
2022/04/06 Python