如何加载保存的DNN估计器模型进行预测？

Question

我有一个已保存的DNN估算器模型。如何还原模型以进行预测？ 我使用以下方式保存了模型：

import time
from selenium import webdriver

chrome_options = webdriver.ChromeOptions()
chrome_options.add_experimental_option('prefs', {'intl.accept_languages': 'es'})
chrome_browser = webdriver.Chrome(options=chrome_options)
chrome_browser.get('https://www.google.es/')
time.sleep(2)  # wait page to load 3 sec
search = chrome_browser.find_element_by_name('q')
search.send_keys('apple pie')
search.submit()

因此，在model_path目录中，我具有：

• 检查点
• event.out.tf ...
• graph.pbtxt
• model.ckpt-1 ...
• model.ckpt-1 ...
• model.ckpt-1 ...
• model.ckpt-5000 ...
• model.ckpt-5000 ...
• model.ckpt-5000 ...

有什么主意吗？

Answer 1

也许您可以按照以下方法尝试predictor。

from tensorflow.contrib import predictor

my_predict = predictor.from_saved_model(model_path)
prediction = my_predict({"x": [your_testing_data]})

希望这会有所帮助！