获取保存的路径_型号.pbML引擎培训后

2024-10-02 08:26:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我一直在使用ML engine的python客户机API来创建一些固定估计器的培训工作。我不能做的是找到被救的人的路_型号.pb因为它存储的路径有一个时间戳作为目录名。不管怎样,我可以在python客户机上使用正则表达式或其他方法来实现这一点,这样我就可以用正确的路径部署模型。你知道吗

路径现在似乎是这样的-

gs://bucket_name/outputs/export/serv/timestamp/saved_model.pb


更新

谢谢你的回答。 所以我写了这个,它给出了我可以传递给ml引擎的deploy\u uri的确切路径。你知道吗

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/')

Tags: name路径client客户机prefixmodelbucketline
2条回答
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)

使用gsutil和tail:

MODEL_LOCATION=$(gsutil ls gs://${BUCKET}/outputs/export/serv | tail -1)
gcloud ml-engine models create ${MODEL_NAME}  regions $REGION
gcloud ml-engine versions create ${MODEL_VERSION}  model ${MODEL_NAME}  origin ${MODEL_LOCATION}  runtime-version $TFVERSION

相关问题 更多 >

    热门问题