[英]How to get predictions of test dataset from a trained model in tensorflow
我已经在 Tensorflow 中训练了一个神经网络模型,并将训练结果保存在一个目录trainlogs
。
在trainlogs
,有一个文件checkpoints
。
我认为这是我必须用来恢复训练参数的文件,但我不确定如何使用tf.train.Saver()
来完成此操作。
这是它的教程: https : //www.tensorflow.org/programmers_guide/saved_model 。
本教程的一个简短片段:
saver = tf.train.Saver()
with tf.Session() as sess:
saver.restore(sess, "/tmp/model.ckpt")
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.