如何从张量流中的训练模型中获得测试数据集的预测

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")