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

时间:2018-03-14 19:01:48

标签: python google-cloud-ml

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

1 个答案:

答案 0 :(得分:2)

针对您的版本资源发出GET请求(docs)将返回一个Version对象,其中包含您要查找的deploymentUri字段。您可以使用任何库以您选择的语言发出HTTP请求,但您需要在标题中发送授权令牌。

这是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']