TF：利用TF的train.Saver將訓練好的W、b模型文件保存+新建載入剛訓練好模型(用于以后預測新的數據)

?

?

目錄

輸出結果

代碼設計

---

?

?

輸出結果

代碼設計

import tensorflow as tf import numpy as npW = tf.Variable([[2,1,8],[1,2,5]], dtype=tf.float32, name='weights') b = tf.Variable([[1,2,5]], dtype=tf.float32, name='biases')init= tf.global_variables_initializer() saver = tf.train.Saver() with tf.Session() as sess: sess.run(init)save_path = saver.save(sess, "niu/save_net.ckpt") print("Save to path: ", save_path) #TF：利用TF的train.Saver載入曾經訓練好的variables(W、b)以供預測新的數據 import tensorflow as tf import numpy as npW = tf.Variable(np.arange(6).reshape((2, 3)), dtype=tf.float32, name="weights") b = tf.Variable(np.arange(3).reshape((1, 3)), dtype=tf.float32, name="biases")saver = tf.train.Saver() with tf.Session() as sess: saver.restore(sess, "niu/save_net.ckpt")print("weights:", sess.run(W))print("biases:", sess.run(b))

?

相關文章
TF：利用TF的train.Saver將訓練好的W、b模型文件保存+新建載入剛訓練好模型(用于以后預測新的數據)

?

總結

以上是生活随笔為你收集整理的TF：利用TF的train.Saver将训练好的W、b模型文件保存+新建载入刚训练好模型(用于以后预测新的数据)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。