如何使用 Lambda 和 API 网关部署由 AWS Sagemaker 创建的乳腺癌预测终端节点？

Question

我正在尝试使用 AWS Lambda 和 API 网关在 Amazon Sagemanker 上部署现有的乳腺癌预测模型。我已经按照下面的官方文档 url.

https://aws.amazon.com/blogs/machine-learning/call-an-amazon-sagemaker-model-endpoint-using-amazon-api-gateway-and-aws-lambda/

我在 "predicted_label" 遇到类型错误。

 result = json.loads(response['Body'].read().decode())
 print(result)
 pred = int(result['predictions'][0]['predicted_label'])
 predicted_label = 'M' if pred == 1 else 'B'

 return predicted_label

如果有人可以解决此问题，请告诉我。谢谢你。

Answer 1

通过print(type(result))打印结果类型，你可以看到它是一个字典。现在您可以看到密钥名称是 "score" 而不是您提供给 pred 的 "predicted_label"。因此将其替换为

pred = int(result['predictions'][0]['score'])

我认为这可以解决您的问题。

这是我的 lambda 函数：

import os
import io
import boto3
import json
import csv

# grab environment variables
ENDPOINT_NAME = os.environ['ENDPOINT_NAME']
runtime= boto3.client('runtime.sagemaker')

def lambda_handler(event, context):
   print("Received event: " + json.dumps(event, indent=2))

   data = json.loads(json.dumps(event))
   payload = data['data']
   print(payload)

   response = runtime.invoke_endpoint(EndpointName=ENDPOINT_NAME,
                                      ContentType='text/csv',
                                      Body=payload)
   #print(response)
   print(type(response))
   for key,value in response.items():
       print(key,value)
   result = json.loads(response['Body'].read().decode())
   print(type(result))
   print(result['predictions'])
   pred = int(result['predictions'][0]['score'])
   print(pred)
   predicted_label = 'M' if pred == 1 else 'B'

   return predicted_label

如何使用 Lambda 和 API 网关部署由 AWS Sagemaker 创建的乳腺癌预测终端节点？

How to deploy breast cancer prediction endpoint created by AWS Sagemaker using Lambda and API gateway?

amazon-web-services

aws-lambda

aws-api-gateway

amazon-sagemaker