tensorflow如何輸出權重值和偏差

這篇文章主要為大家展示了“tensorflow如何輸出權重值和偏差”，內容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領大家一起研究并學習一下“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=",")

以上是“tensorflow如何輸出權重值和偏差”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！