我想对通过ML引擎训练的保存模型进行本地预测。对于我的CNN模型,我使用的是花卉数据集(https://github.com/GoogleCloudPlatform/cloudml-samples/tree/master/flowers)的示例代码。
如何恢复我的模型并做出预测,以返回具有准确度得分的列表或字典?
我已经尝试恢复我的模型,但是没有成功。
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访问预测分数。