无法使用 AWS-Lambda 调用在 sagemaker 中创建的 XG-Boost 端点

Question

我在 AWS-Sagemaker 上训练了一个 xgboost 模型并创建了一个端点。现在我想使用 AWS Lambda 和 AWS API 调用端点。我创建了一个 lambda 函数并为我的 xgboost 模型添加了下面提到的代码。当我尝试对其进行测试时，该函数抛出一个 ParamValidation 错误。这是我的代码

import json
import os
import csv
import io
import boto3
endpointname =os.environ['endpointname'] #name of the endpoint I created in sagemaker
runtime = boto3.client('runtime.sagemaker')
def lambda_handler(event, context):
    print("Recieved Event: "+json.dumps(event,indent=2))
    data=json.loads(json.dumps(event))
    print(data)
    response = runtime.invoke_endpoint(EndpointName=endpointname,ContentType='text/csv',Body=data)
    print(response)
    result = json.loads(response['Body'].read().decode())
    print(int(float(result))) #sagemaker xgb returns bytes type for the test case

我创建的测试事件是dict类型。该函数正在抛出 Invalid type for parameter Body, value: {'Time':'7'}, type: <class 'dict'>, valid types: <class 'bytes'>, <class 'bytearray'>, file-like object 这意味着我应该将 byte 或 bytearray 而不是 dict 类型传递到我的事件中。但是当我阅读这篇 AWS Lambda doc It says that my event type can only be dict,int,list,float,str, or None type. I followed the steps mentioned in this aws 文档来创建我的 lambda 函数时。有人可以解释为什么我的代码会抛出上述错误吗？

Answer 1

data=json.loads(json.dumps(event)) 是多余的操作。 data=event 将 return True。我们为测试用例提供的事件是dict类型。它有一个键值对。键可以是任何东西，值应该是由逗号分隔的所有预测变量的单个字符串。为了预测输出，我们需要测试用例的值。例如，声明 payload=data['key'] 然后在 response 中更改 Body=payload。然后就可以了

无法使用 AWS-Lambda 调用在 sagemaker 中创建的 XG-Boost 端点

Unable to call XG-Boost endpoint created in sagemaker using AWS-Lambda

python

machine-learning

amazon-web-services

aws-lambda

amazon-sagemaker