2017-03-05 92 views
1

我培养出“递归神经网络”是继tutorial来预测下一个单词,并得到了一个模型:使用模型中使用Tensorflow

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(“下一个单词是什么?”)的函数来返回预测的单词。

回答

2

您想使用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

+0

我没有与CKPT扩展文件。 – Ariel