使用lambda函数调用sagemaker端点(call sagemaker endpoint us

2019-10-29 15:09发布

我在S3的一些数据,我想创建一个lambda函数预测的输出与我的部署AWS sagemaker端点,然后我把输出的S3试。 是否有必要在这种情况下创建一个API网关像描述下这个链接 ? 而在lambda函数什么,我必须把。 我希望把(在哪里可以找到数据,如何调用终端,其中把数据)

谢谢

Answer 1:

你绝对不必创建API网关的API。 您可以直接使用invoke_endpoint()API,通过端点名称,内容类型,有效载荷调用的端点。

例如:

import boto3

endpoint_name = <INSERT_ENDPOINT_NAME>
runtime = boto3.Session().client(service_name='sagemaker-runtime',region_name='us-east-1')

response = runtime.invoke_endpoint(EndpointName=endpoint_name, ContentType='application/x-image', Body=payload)
print(response['Body'].read())

这里更多示例使用Lambda函数: https://medium.com/@julsimon/using-chalice-to-serve-sagemaker-predictions-a2015c02b033



文章来源: call sagemaker endpoint using lambda function