[英]How to load .ckpt, .meta output ffiles rom a TF model, input prediction data and make a prediction?
我按照教程使用 tf.estimator.DNNLinearCombinedRegressor 来训练 TF model 来预测芝加哥出租车费用。 model 在 Google Cloud Storage 存储桶中生成以下输出。
问题:如何使用这些文件加载 model,输入新数据点并进行预测? 假设这些是变量和对应的值:
我在网上搜索了答案,这就是我现在所拥有的。 它给了我错误信息。 任何人都可以帮助并让我知道如何纠正它吗? 谢谢!
with tf.Session() as sess:
new_saver = tf.train.import_meta_graph('gs://folder_name/model.ckpt-15165.meta')
new_saver.restore(sess, tf.train.latest_checkpoint('gs://folder_name/'))
graph = tf.get_default_graph()
x = graph.get_tensor_by_name("Placeholder:0")
feed_dict = {x:[8, 41.5, -87.2, 41.9, -87.1]}
print(sess.run(feed_dict))
我想到了。
内置 function,便于预测。
pred_input_func = tf.estimator.inputs.pandas_input_fn(x = x_test, shuffle = False)
predictions = estimator.predict(pred_input_func)
for result in predictions:
print (result)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.