Question

我训练了一个＆＃34;回归神经网络＆＃34;在tutorial之后得到一个模型：

model/
model/model.ckpt-72985.data-00000-of-00001
model/model.ckpt-72883.meta
model/model.ckpt-72793.meta
model/model.ckpt-72985.meta
model/model.ckpt-72793.index
model/model.ckpt-72971.data-00000-of-00001
model/events.out.tfevents.1488038213.mymachine
model/model.ckpt-72883.index
model/model.ckpt-72985.index
model/graph.pbtxt
model/model.ckpt-72883.data-00000-of-00001
model/model.ckpt-72793.data-00000-of-00001
model/model.ckpt-72971.index
model/model.ckpt-72971.meta
model/checkpoint

现在，我想用这个模型来预测程序中的下一个单词，我该怎么办？

我想使用诸如nextWord之类的函数（＆＃34;下一个字是什么？＆＃34;）来返回预测的单词。

Answer 1

您想使用tf.train.Saver。

saved_model = tf.train.get_checkpoint_state('./model/') #or your model path
saver = tf.train.Saver()

with tf.Session() as sess:
    saver.restore(sess, saved_model.model_checkpoint_path)

另外，您可能需要查看this example

使用模型使用Tensorflow预测下一个单词

1 个答案: