如何從張量流中的訓練模型中獲得測試數據集的預測

Question

我已經在Tensorflow中訓練了一個神經網絡模型並將訓練結果保存在目錄trainlogs中。

在trainlogs之內，有一個文件checkpoints。

我認爲這是我必須使用的文件來恢復訓練的參數，但我不知道如何使用tf.train.Saver()來完成此操作。

Answer 1

這是一個教程：https://www.tensorflow.org/programmers_guide/saved_model。

從本教程的簡短片段：

saver = tf.train.Saver() 
with tf.Session() as sess: 
    saver.restore(sess, "/tmp/model.ckpt")