tensorflow輸出權(quán)重值和偏差的方法
更新時間：2018年02月10日 16:39:05 作者：林先生您好
本篇文章主要介紹了tensorflow輸出權(quán)重值和偏差的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
使用tensorflow 訓(xùn)練模型時,我們可以使用 tensorflow自帶的 Save模塊 tf.train.Saver()來保存模型,使用方式很簡單 就是在訓(xùn)練完模型后,調(diào)用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)
但是這種方式保存的模型中包含特別多的信息,使保存的模型很大,其實里面有很多不是我們想要的.我們就想要里面最重要的權(quán)重信息和偏差等等數(shù)據(jù),然后再自己寫解密代碼,就可以把模型應(yīng)用于其他的平臺,比如安卓手機(jī).
那么我們可以使用下面的方式獲取訓(xùn)練后的權(quán)重和偏移,
ww, bb = sess.run([self.W,self.b])
其中W,和b都是 Tensor類型的數(shù)據(jù)
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 輸出權(quán)重 到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=",")
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python中字符串的常見操作技巧,結(jié)合實例形式總結(jié)分析了Python針對字符串的反轉(zhuǎn)、編碼轉(zhuǎn)換、進(jìn)制轉(zhuǎn)換及運算等常用技巧,需要的朋友可以參考下2016-07-07
這篇文章介紹了python實現(xiàn)C++代碼生成器的方法，文中通過示例代碼介紹的非常詳細(xì)。對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-07-07
平常我們想給某些圖片添加文字水印,方法有很多,也有很多的工具可以方便的進(jìn)行,今天主要是對PIL庫的應(yīng)用,結(jié)合Python語言批量對圖片添加水印,文章通過代碼示例給大家介紹的非常詳細(xì),感興趣的同學(xué)可以參考一下2023-11-11
這篇文章主要介紹了Selenium自動化測試實現(xiàn)窗口切換，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
今天小編就為大家分享一篇python3讀取csv和xlsx文件的實例，具有很好的參考價值，希望對的大家有所幫助。一起跟隨小編過來看看吧2018-06-06
今天小編就為大家分享一篇關(guān)于Python中按鍵來獲取指定的值，小編覺得內(nèi)容挺不錯的，現(xiàn)在分享給大家，具有很好的參考價值，需要的朋友一起跟隨小編來看看吧2019-03-03
Django模版引擎中最強(qiáng)大也是最復(fù)雜的部分就是模版繼承了，頁面的代碼很多，隨隨便便就是幾百行代碼，但是每個頁面之中都有相同的元素。本文主要介紹了Django模板的繼承及引用，感興趣的可以了解一下2021-08-08
本文主要是總結(jié)yaml常用的一些數(shù)據(jù)形式，對于我們正常的項目使用，已經(jīng)足夠用了，接下來通過本文給大家分享python 中yaml文件用法，需要的朋友參考下吧2021-07-07
linux下python使用sendmail發(fā)送郵件
這篇文章主要為大家詳細(xì)介紹了linux下python使用sendmail發(fā)送郵件，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-05-05
字符串是所有編程語言中都有的基本變量的類型?,程序員基本每天都在和字符串打交道,下面這篇文章主要給大家介紹了關(guān)于Python字符串拼接的4種方法,文中通過實例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-07-07