部署模型

Question

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

Trained model in the direction Train

Data in the directory Data

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

任何答案将不胜感激。

Answer 1

因此，在获得训练有素的模型的文件夹中，只需要将其加载到会话中即可。首先创建一个保护程序（也可以将其用于下载）

train_saver = tf.train.Saver()

现在进入您的会话：

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

然后只用feed_dict来输入张量。

其他选择是创建一个protobuf文件（.pb），但这样做时，您将必须按照我说的那样加载模型。