tensorflow - 部署模型

Question

我刚刚按照https://github.com/GoogleCloudPlatform/MiniCat中描述的方式完成了分类器模型的训练，但我不确定如何使用该模型进行预测。

方向训练的训练模型

目录数据中的数据

我对此真的很陌生，我不知道从哪里开始。我在https://cloud.google.com/ml-engine/docs/tensorflow/deploying-models中阅读了一些关于部署模型的内容，但我什至如何创建 SavedModel。

任何答案将不胜感激。

标签： tensorflowgoogle-cloud-platform

Answer 1

因此，在您获得训练模型的文件夹中，您只需在会话中加载该模型。首先创建一个saver（你也可以用它来做laoding）

train_saver = tf.train.Saver()

现在在您的会话中：

train_saver.restore(sess, 'path/to/model/doc_classifier_cnn_model.ckpt')

然后用 feed_dict 喂给张量。

其他选择是创建一个 protobuf 文件 (.pb)，但这样做你必须加载模型，如我所说。