ai.djl.aws.sagemaker

Class SageMaker

java.lang.Object

ai.djl.aws.sagemaker.SageMaker

public final class SageMaker
extends java.lang.Object

A utility class that help deploy model to SageMaker.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SageMaker.Builder A Builder to construct a SageMaker.

Method Summary

Modifier and Type	Method and Description
static SageMaker.Builder	builder() Creates a builder to build a SageMaker instance.
void	deleteEndpoint() Deletes the Amazon SageMaker endpoint.
void	deleteEndpointConfig() Deletes the endpoint configuration.
void	deleteSageMakerModel() Deletes the SageMaker model configuration.
void	deploy() Deploys the model to Amazon SageMaker hosting service.
byte[]	invoke(byte[] body) Invokes the Amazon SageMaker endpoint.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

deploy

public void deploy()
            throws java.io.IOException

Deploys the model to Amazon SageMaker hosting service.

Throws:

java.io.IOException - if failed upload model to S3

deleteEndpoint

public void deleteEndpoint()

Deletes the Amazon SageMaker endpoint.

deleteEndpointConfig

public void deleteEndpointConfig()

Deletes the endpoint configuration.

deleteSageMakerModel

public void deleteSageMakerModel()

Deletes the SageMaker model configuration.

invoke

public byte[] invoke(byte[] body)

Invokes the Amazon SageMaker endpoint.

Parameters:

body - the request payload

Returns:

the inference response

builder

public static SageMaker.Builder builder()

Creates a builder to build a SageMaker instance.

Returns:

a new builder