pandas - 调用 sagemaker 端点时如何构建 "text/csv"有效负载

Question

我的训练数据看起来像

df = pd.DataFrame({'A' : [2, 5], 'B' : [1, 7]})

我在 AWS Sagemaker 中训练了一个模型，并在端点后面部署了该模型。
端点接受负载为“text/csv”。

要使用 boto3 调用端点，您可以执行以下操作:

import boto3
client = boto3.client('sagemaker-runtime')
response = client.invoke_endpoint(
    EndpointName="my-sagemaker-endpoint-name",
    Body= my_payload_as_csv,
    ContentType = 'text/csv')

我如何从我的数据帧构建有效负载“my_payload_as_csv”以便正确调用 Sagemaker 端点？

Answer 1

如果您从数据框示例开始

df = pd.DataFrame({'A' : [2, 5], 'B' : [1, 7]})

你排

df_1_record = df[:1]

并转换 df_1_record到这样的csv:

import io
from io import StringIO
csv_file = io.StringIO()
# by default sagemaker expects comma seperated
df_1_record.to_csv(csv_file, sep=",", header=False, index=False)
my_payload_as_csv = csv_file.getvalue()

my_payload_as_csv好像

'2,1\n'

然后您可以调用 sagemaker 端点

import boto3
client = boto3.client('sagemaker-runtime')
response = client.invoke_endpoint(
    EndpointName="my-sagemaker-endpoint-name",
    Body= my_payload_as_csv,
    ContentType = 'text/csv')