amazon-sagemaker - 如何使用 sagemaker java API 调用端点？

Question

我试图运行这个例子:tensorflow_abalone_age_predictor_using_layers，其中 abalone_predictor.predict(tensor_proto) 用于调用端点并进行预测。我试图使用 java API AmazonSageMakerRuntime 来实现相同的效果，但我不知道如何指定 body 和 contentType InvokeEndPointRequest。该文件没有详细说明请求的格式。非常感谢任何帮助!

Answer 1

我没有尝试过具体的例子，但下面的代码片段应该可以帮助您调用端点进行预测

InvokeEndpointRequest invokeEndpointRequest = new InvokeEndpointRequest();
invokeEndpointRequest.setContentType("application/x-image");
ByteBuffer buf = ByteBuffer.wrap(image);

invokeEndpointRequest.setBody(buf);
invokeEndpointRequest.setEndpointName(endpointName);
invokeEndpointRequest.setAccept("application/json");

AmazonSageMakerRuntime amazonSageMaker = AmazonSageMakerRuntimeClientBuilder.defaultClient();
InvokeEndpointResult invokeEndpointResult = amazonSageMaker.invokeEndpoint(invokeEndpointRequest);

我看到您正在尝试创建 TensorProto 并传递给端点请求的示例。你可以试试create a TensorProto您的调用请求并设置为正文