tensorflow输出权重值和偏差的方法
更新时间：2018年02月10日 16:39:05 作者：林先生您好
本篇文章主要介绍了tensorflow输出权重值和偏差的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
使用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实现这一游戏，感兴趣的可以尝试一下2022-08-08
这篇文章主要介绍了python脚本实现查找webshell的方法,是很实用的一个功能,需要的朋友可以参考下2014-07-07
今天小编就为大家分享一篇对python 中class与变量的使用方法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-06-06
queue是python的标准库，俗称队列.可以直接import引用,在python2.x中,模块名为Queue。这篇文章主要介绍了python爬虫中多线程的使用,需要的朋友可以参考下2019-09-09
本文详细介绍了在已安装Python的情况下如何正确安装和配置Anaconda环境,包括Python环境变量的配置、Anaconda的下载安装以及环境变量配置,文中通过代码就介绍的非常详细,需要的朋友可以参考下2024-11-11
pycharm远程连接服务器调试tensorflow无法加载问题
最近打算在win系统下使用pycharm开发程序，并远程连接服务器调试程序，其中在import tensorflow时报错，本文就来介绍一下如何解决，感兴趣的可以了解一下2021-06-06
本文主要介绍了Python Unittest ddt数据驱动的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-02-02
这篇文章主要介绍了Python3 selenium 实现QQ群接龙自动化功能,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-04-04
今天小编就为大家分享一篇Python 使用PIL中的resize进行缩放的实例讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-08-08
Python 详解爬取并统计CSDN全站热榜标题关键词词频流程
读万卷书不如行万里路，只学书上的理论是远远不够的，只有在实战中才能获得能力的提升，本篇文章手把手带你用Python爬取CSDN全站综合热榜标题，顺便统计关键词词频,大家可以在过程中查缺补漏，提升水平2021-11-11
最新评论