從Google Cloud Datalab筆記本下載保存的模型

Question

我是Google Cloud Platform的新手。我已將自己的代碼寫入Cloud Datalab，並在使用數據進行訓練後將模型保存在datalabs筆記本中。我想在本地機器上下載保存的模型並將其用於進一步預測。我在本地機器上保存了模型，並試圖用它運行預測。但是我得到一個錯誤，說「NotFittedError：無法找到訓練好的模型」。誰能告訴我如何解決這個問題？

Answer 1

你能告訴我你如何做預測嗎？通過gcloud？另外，你如何保存模型？你打電話給TensorFlow的tf.python.saved_model庫來保存它嗎？

使用gcloud，可以通過目錄指定模型。該目錄應該包含一個「saved_model.pb」文件和一些其他目錄。您可以檢查模型目錄以查看「saved_model.pb」是否存在。