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