在ML引擎训练后获取saved_model.pb的路径

时间:2018-03-18 11:41:20

标签: python tensorflow google-cloud-storage google-cloud-ml

我一直在使用ML引擎的python客户端API来创建一些罐头估算器的培训工作。我无法做的是在GCS上获取saved_model.pb的路径,因为它存储的路径有一个时间戳作为目录名称。无论如何我可以使用正则表达式或python客户端上的某些东西来获取它,这样我就能够使用正确的路径部署模型。

此路径似乎现在采用这种格式 -

  

GS:// BUCKET_NAME /输出/出口/ SERV / timestamp /saved_model.pb

更新

感谢shahin的回答。 所以我写了这个,它给了我确切的路径,我可以传递给deploy_uri为ml引擎。

from google.cloud import storage

def getGCSPath(prefix):
    storage_client = storage.Client()
    bucket = storage_client.get_bucket(bucket_name)
    mlist = bucket.list_blobs(prefix=prefix)
    for line in mlist:
        if 'saved_model.pb' in line.name:
            return line.name[:-14]

# print getGCSPath('output/export/serv/')

2 个答案:

答案 0 :(得分:1)

使用gsutil和tail:

path/to/svg_file.svg#remove-alpha

答案 1 :(得分:0)

import os
import cloudstorage as gcs
bucket = os.environ.get('BUCKET')
page_size = 1
stats = gcs.listbucket(bucket + '/outputs/export/serv', max_keys=page_size)