从 Google Cloud Datalab 笔记本下载保存的模型

Question

我是 Google Cloud Platform 的新手。我已经在 Cloud Datalab 中编写了代码，并在使用数据对其进行训练后将模型保存在 datalabs 笔记本中。我想在我的本地机器上下载保存的模型，并用它来做进一步的预测。我将模型保存在本地机器中并尝试使用它进行预测。但我收到一条错误消息，提示“NotFittedError: Couldn't find trained model at”。谁能告诉我如何解决这个问题？

Answer 1

你能告诉我你是怎么做预测的吗？通过gcloud？另外，如何保存模型？调用 TensorFlow 的 tf.python.saved_model 库来保存吗？

使用 gcloud，您可以通过目录指定模型。该目录应该包括一个“saved_model.pb”文件和一些其他目录。您可以检查模型目录以查看“saved_model.pb”是否存在。