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采用socket模拟TCP通讯的实现方法
Nov 19 Python
Python命令行参数解析模块optparse使用实例
Apr 13 Python
python实现发送和获取手机短信验证码
Jan 15 Python
Python给你的头像加上圣诞帽
Jan 04 Python
python3+PyQt5实现文档打印功能
Apr 24 Python
python RabbitMQ 使用详细介绍(小结)
Nov 08 Python
python基于paramiko将文件上传到服务器代码实现
Jul 08 Python
python 多进程队列数据处理详解
Dec 23 Python
tensorflow中tf.slice和tf.gather切片函数的使用
Jan 19 Python
python实现人机猜拳小游戏
Feb 03 Python
基于Python采集爬取微信公众号历史数据
Nov 27 Python
浅谈Python中的函数(def)及参数传递操作
May 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编程最快明白》第五讲:php目录、文件操作
2010/11/01 PHP
php 表单提交大量数据发生丢失的解决方法
2014/03/03 PHP
Codeigniter实现发送带附件的邮件
2015/03/19 PHP
php通过排列组合实现1到9数字相加都等于20的方法
2015/08/03 PHP
PHP中的session安全吗?
2016/01/22 PHP
PHP生成和获取XML格式数据的方法
2016/03/04 PHP
PHP命名空间简单用法示例
2018/12/28 PHP
PHP的图像处理实例小结【文字水印、图片水印、压缩图像等】
2019/12/20 PHP
优化Node.js Web应用运行速度的10个技巧
2014/09/03 Javascript
javascript中Date对象的getDay方法使用指南
2014/12/22 Javascript
Nodejs学习笔记之Global Objects全局对象
2015/01/13 NodeJs
JQ实现新浪游戏首页幻灯片
2015/07/29 Javascript
javascript实现unicode与ASCII相互转换的方法
2015/12/10 Javascript
JS+CSS实现下拉刷新/上拉加载插件
2017/03/31 Javascript
3分钟快速搭建nodejs本地服务器方法运行测试html/js
2017/04/01 NodeJs
利用JS做网页特效_大图轮播(实例讲解)
2017/08/09 Javascript
jquery动态添加以及遍历option并获取特定样式名称的option方法
2018/01/29 jQuery
layer.prompt使文本框为空的情况下也能点击确定的方法
2019/09/24 Javascript
解决vue 表格table列求和的问题
2019/11/06 Javascript
Python 异常处理实例详解
2014/03/12 Python
Python基础入门之seed()方法的使用
2015/05/15 Python
Python有序查找算法之二分法实例分析
2017/12/11 Python
python并发编程之线程实例解析
2017/12/27 Python
Python列表如何更新值
2020/05/27 Python
基于html5绘制圆形多角图案
2016/04/21 HTML / CSS
屈臣氏官方旗舰店:亚洲享负盛名的保健及美妆零售商
2019/03/15 全球购物
应届生船舶驾驶求职信
2013/10/19 职场文书
大气污染防治方案
2014/05/19 职场文书
图书馆志愿者活动总结
2014/06/27 职场文书
机械专业求职信范文
2014/07/15 职场文书
2014大四本科生自我鉴定总结
2014/10/04 职场文书
党员考试作弊检讨书1000字
2015/02/16 职场文书
小区环境卫生倡议书
2015/04/29 职场文书
2015国庆节放假通知范文
2015/07/30 职场文书
导游词之张家口
2019/12/13 职场文书
html form表单基础入门案例讲解
2021/07/15 HTML / CSS