从 jupyter notebook 中获取特定 GCMLE 模型的 saved_model.pb 位置

Question

我将 GCMLE 模型部署为 model 用于 GCMLE 预测服务。在 GCP 控制台 UI 中，我可以导航到模型名称 -> 版本，然后我可以查看 Model location（即 gs://..../），它指定了我的 saved_model.pb 文件。有什么办法可以在 jupyter notebook/python 脚本中动态获取这个 "model location" 吗？然后，我希望能够使用此模型位置在本地下载 saved_model.pb，以便我可以将模型加载到本地会话中，以便在特定推理任务上进行调试。目前，我可以手动完成所有这些操作，但我希望能够在模型版本之间快速切换，而无需手动 track/download saved_model.pb 文件。

Answer 1

正在发出 GET 请求 (docs) against your version resource will return a Version object，其中包含您要查找的 deploymentUri 字段。您可以使用任何库以您选择的语言发出 HTTP 请求，但您确实需要在 headers.

中发送授权令牌

这是 Python 中的示例：

import requests
from oauth2client.client import GoogleCredentials

token = GoogleCredentials.get_application_default().get_access_token().access_token
api = 'https://ml.googleapis.com/v1/projects/MYPROJECT/models/MYMODEL/versions/MYVERSION'
headers = {'Authorization': 'Bearer ' + token }    
response = requests.get(api, headers=headers)
print response.json()['deploymentUri']

从 jupyter notebook 中获取特定 GCMLE 模型的 saved_model.pb 位置

Obtain saved_model.pb location for specific GCMLE model from within jupyter notebook

python

google-cloud-ml