ML引擎模型的本地预测

时间:2018-08-21 13:18:29

标签: tensorflow google-cloud-platform tensorflow-serving google-cloud-ml

我想对通过ML引擎训练的保存模型进行本地预测。对于我的CNN模型,我使用的是花卉数据集(https://github.com/GoogleCloudPlatform/cloudml-samples/tree/master/flowers)的示例代码。

我有以下文件: enter image description here

如何恢复我的模型并做出预测,以返回具有准确度得分的列表或字典?

我已经尝试恢复我的模型,但是没有成功。

import tensorflow as tf

with tf.Session() as sess:    
    saver = tf.train.import_meta_graph('saved_model.pb')
    saver.restore(sess,tf.train.latest_checkpoint('variables/'))
    print(sess.run('w1:0'))

注1:我的目标是制作分类报告或/和混淆矩阵,就像在Keras中一样。

注2:我的模型由87个标签组成。

注3:已通过ML Engine的在线预测API访问预测分数。

0 个答案:

没有答案