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写一个贪吃蛇游戏实例代码
Aug 21 Python
Python基础语言学习笔记总结(精华)
Nov 14 Python
Python中Scrapy爬虫图片处理详解
Nov 29 Python
Python3实现发送QQ邮件功能(html)
Dec 15 Python
关于python2 csv写入空白行的问题
Jun 22 Python
pygame游戏之旅 调用按钮实现游戏开始功能
Nov 21 Python
详解Python传入参数的几种方法
May 16 Python
Python实现最常见加密方式详解
Jul 13 Python
python爬虫 正则表达式解析
Sep 28 Python
Python pickle模块实现对象序列化
Nov 22 Python
Tensorflow累加的实现案例
Feb 05 Python
Django ModelForm操作及验证方式
Mar 30 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
基于simple_html_dom的使用小结
2013/07/01 PHP
PHP使用mongoclient简单操作mongodb数据库示例
2019/02/08 PHP
PHP使用HTML5 FileApi实现Ajax上传文件功能示例
2019/07/01 PHP
简单的php购物车代码
2020/06/05 PHP
图片自动缩小 点击放大
2008/07/07 Javascript
ExtJS的拖拽效果示例
2013/12/09 Javascript
js校验表单后提交表单的三种方法总结
2014/02/28 Javascript
js获取客户端网卡的IP地址、MAC地址
2014/03/26 Javascript
jquery.validate.js插件使用经验记录
2014/07/02 Javascript
javascript模拟评分控件实现方法
2015/05/13 Javascript
jQuery+formdata实现上传进度特效遇到的问题
2016/02/24 Javascript
jQuery实现简洁的轮播图效果实例
2016/09/07 Javascript
JavaScript学习笔记整理_简单实现枚举类型,扑克牌应用
2016/09/19 Javascript
利用Query+bootstrap和js两种方式实现日期选择器
2017/01/10 Javascript
用Vue.extend构建消息提示组件的方法实例
2017/08/08 Javascript
NodeJS设计模式总结【单例模式,适配器模式,装饰模式,观察者模式】
2017/09/06 NodeJs
基于vue-ssr的静态网站生成器VuePress 初体验
2018/04/17 Javascript
linux 后台运行node服务指令方法
2018/05/23 Javascript
jQuery实现的简单手风琴效果示例
2018/08/29 jQuery
nuxt.js中间件实现拦截权限判断的方法
2018/11/21 Javascript
JavaScript Image对象实现原理实例解析
2020/08/26 Javascript
vue 子组件和父组件传值的示例
2020/09/11 Javascript
vue 避免变量赋值后双向绑定的操作
2020/11/07 Javascript
使用Python的Tornado框架实现一个Web端图书展示页面
2016/07/11 Python
完美解决Python 2.7不能正常使用pip install的问题
2018/06/12 Python
Django1.11配合uni-app发起微信支付的实现
2019/10/12 Python
蹦床仓库:Trampoline Warehouse
2018/12/06 全球购物
I.T集团香港官方商城:ITeSHOP.com Hong Kong
2019/02/15 全球购物
工作态度恶劣检讨书
2015/05/06 职场文书
运动会报道稿大全
2015/07/23 职场文书
搞笑婚礼主持词开场白
2015/11/24 职场文书
大学生社区义工服务心得体会
2016/01/22 职场文书
2016年“12.3”国际残疾人日活动总结
2016/04/01 职场文书
MySQL query_cache_type 参数与使用详解
2021/07/01 MySQL
深入讲解数据库中Decimal类型的使用以及实现方法
2022/02/15 MySQL
Golang 并发编程 SingleFlight模式
2022/04/26 Golang