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)用于其他的平臺,比如安卓手機.
那么我們可以使用下面的方式獲取訓(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實現(xiàn)生命游戲的示例代碼(tkinter版)
生命游戲是由劍橋大學(xué)約翰·何頓·康威設(shè)計的計算機程序，一時吸引了各行各業(yè)一大批人的興趣。本文將用Python實現(xiàn)這一游戲，感興趣的可以嘗試一下2022-08-08
這篇文章主要介紹了python腳本實現(xiàn)查找webshell的方法,是很實用的一個功能,需要的朋友可以參考下2014-07-07
今天小編就為大家分享一篇對python 中class與變量的使用方法詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
queue是python的標準庫，俗稱隊列.可以直接import引用,在python2.x中,模塊名為Queue。這篇文章主要介紹了python爬蟲中多線程的使用,需要的朋友可以參考下2019-09-09
本文詳細介紹了在已安裝Python的情況下如何正確安裝和配置Anaconda環(huán)境,包括Python環(huán)境變量的配置、Anaconda的下載安裝以及環(huán)境變量配置,文中通過代碼就介紹的非常詳細,需要的朋友可以參考下2024-11-11
pycharm遠程連接服務(wù)器調(diào)試tensorflow無法加載問題
最近打算在win系統(tǒng)下使用pycharm開發(fā)程序，并遠程連接服務(wù)器調(diào)試程序，其中在import tensorflow時報錯，本文就來介紹一下如何解決，感興趣的可以了解一下2021-06-06
Python Unittest ddt數(shù)據(jù)驅(qū)動的實現(xiàn)
本文主要介紹了Python Unittest ddt數(shù)據(jù)驅(qū)動的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02
Python3 selenium 實現(xiàn)QQ群接龍自動化功能
這篇文章主要介紹了Python3 selenium 實現(xiàn)QQ群接龍自動化功能,本文通過實例代碼給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-04-04
今天小編就為大家分享一篇Python 使用PIL中的resize進行縮放的實例講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-08-08
Python 詳解爬取并統(tǒng)計CSDN全站熱榜標題關(guān)鍵詞詞頻流程
讀萬卷書不如行萬里路，只學(xué)書上的理論是遠遠不夠的，只有在實戰(zhàn)中才能獲得能力的提升，本篇文章手把手帶你用Python爬取CSDN全站綜合熱榜標題，順便統(tǒng)計關(guān)鍵詞詞頻,大家可以在過程中查缺補漏，提升水平2021-11-11