如何恢复使用Estimator API构建的预训练模型?

时间:2018-03-02 02:56:30

标签: model restore

提前致谢。

我是tensorflow的新手,并按照本教程训练此模型。 https://www.tensorflow.org/tutorials/layers我想在另一个地方恢复此模型以获得预测。但我找不到办法。已经工作了两天,仍然不知道该怎么做。如果有人能给我一些建议,我将非常感激。 我尝试了以下内容。

saver=tf.train.import_meta_graph('./model.meta')
saver.restore(sess,tf.train.latest_checkpoint('./')

我认为上面这些我恢复了我的模特?但是我怎样才能进行操作? 如果我问愚蠢的问题,请不要开枪。

1 个答案:

答案 0 :(得分:0)

估算器分类器有预测 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/estimator/estimator.py

predict_input_fn=tf.estimator.inputs.numpy_input_fn(x={"x":x_value},batch_size=1,shuffle=False)
predict_result=mnist_classifier.predict(input_fn=predict_input_fn,predict_keys="classes")
然后问题就解决了。^ ^