software.amazon.awssdk.services.sagemaker.model

Class InputConfig

java.lang.Object

software.amazon.awssdk.services.sagemaker.model.InputConfig

All Implemented Interfaces:

Serializable, SdkPojo, ToCopyableBuilder<InputConfig.Builder,InputConfig>

@Generated(value="software.amazon.awssdk:codegen")
public final class InputConfig
extends Object
implements SdkPojo, Serializable, ToCopyableBuilder<InputConfig.Builder,InputConfig>

Contains information about the location of input model artifacts, the name and shape of the expected data inputs, and the framework in which the model was trained.

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static interface	InputConfig.Builder

Method Summary

Modifier and Type	Method and Description
static InputConfig.Builder	builder()
String	dataInputConfig() Specifies the name and shape of the expected data inputs for your trained model with a JSON dictionary form.
boolean	equals(Object obj)
boolean	equalsBySdkFields(Object obj)
Framework	framework() Identifies the framework in which the model was trained.
String	frameworkAsString() Identifies the framework in which the model was trained.
<T> Optional<T>	getValueForField(String fieldName, Class<T> clazz)
int	hashCode()
String	s3Uri() The S3 path where the model artifacts, which result from model training, are stored.
List<SdkField<?>>	sdkFields()
static Class<? extends InputConfig.Builder>	serializableBuilderClass()
InputConfig.Builder	toBuilder()
String	toString() Returns a string representation of this object.

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface software.amazon.awssdk.utils.builder.ToCopyableBuilder

copy

Method Detail

s3Uri

public String s3Uri()

The S3 path where the model artifacts, which result from model training, are stored. This path must point to a single gzip compressed tar archive (.tar.gz suffix).

Returns:

The S3 path where the model artifacts, which result from model training, are stored. This path must point to a single gzip compressed tar archive (.tar.gz suffix).

dataInputConfig

public String dataInputConfig()

Specifies the name and shape of the expected data inputs for your trained model with a JSON dictionary form. The data inputs are InputConfig$Framework specific.

• TensorFlow: You must specify the name and shape (NHWC format) of the expected data inputs using a dictionary format for your trained model. The dictionary formats required for the console and CLI are different.

  • Examples for one input:

    • If using the console, {"input":[1,1024,1024,3]}

    • If using the CLI, {\"input\":[1,1024,1024,3]}

  • Examples for two inputs:

    • If using the console, {"data1": [1,28,28,1], "data2":[1,28,28,1]}

    • If using the CLI, {\"data1\": [1,28,28,1], \"data2\":[1,28,28,1]}

• KERAS: You must specify the name and shape (NCHW format) of expected data inputs using a dictionary format for your trained model. Note that while Keras model artifacts should be uploaded in NHWC (channel-last) format, DataInputConfig should be specified in NCHW (channel-first) format. The dictionary formats required for the console and CLI are different.

  • Examples for one input:

    • If using the console, {"input_1":[1,3,224,224]}

    • If using the CLI, {\"input_1\":[1,3,224,224]}

  • Examples for two inputs:

    • If using the console, {"input_1": [1,3,224,224], "input_2":[1,3,224,224]}

    • If using the CLI, {\"input_1\": [1,3,224,224], \"input_2\":[1,3,224,224]}

• MXNET/ONNX: You must specify the name and shape (NCHW format) of the expected data inputs in order using a dictionary format for your trained model. The dictionary formats required for the console and CLI are different.

  • Examples for one input:

    • If using the console, {"data":[1,3,1024,1024]}

    • If using the CLI, {\"data\":[1,3,1024,1024]}

  • Examples for two inputs:

    • If using the console, {"var1": [1,1,28,28], "var2":[1,1,28,28]}

    • If using the CLI, {\"var1\": [1,1,28,28], \"var2\":[1,1,28,28]}

• PyTorch: You can either specify the name and shape (NCHW format) of expected data inputs in order using a dictionary format for your trained model or you can specify the shape only using a list format. The dictionary formats required for the console and CLI are different. The list formats for the console and CLI are the same.

  • Examples for one input in dictionary format:

    • If using the console, {"input0":[1,3,224,224]}

    • If using the CLI, {\"input0\":[1,3,224,224]}

  • Example for one input in list format: [[1,3,224,224]]

  • Examples for two inputs in dictionary format:

    • If using the console, {"input0":[1,3,224,224], "input1":[1,3,224,224]}

    • If using the CLI, {\"input0\":[1,3,224,224], \"input1\":[1,3,224,224]}

  • Example for two inputs in list format: [[1,3,224,224], [1,3,224,224]]

• XGBOOST: input data name and shape are not needed.

Returns:

Specifies the name and shape of the expected data inputs for your trained model with a JSON dictionary form. The data inputs are InputConfig$Framework specific.

• TensorFlow: You must specify the name and shape (NHWC format) of the expected data inputs using a dictionary format for your trained model. The dictionary formats required for the console and CLI are different.

  • Examples for one input:

    • If using the console, {"input":[1,1024,1024,3]}

    • If using the CLI, {\"input\":[1,1024,1024,3]}

  • Examples for two inputs:

    • If using the console, {"data1": [1,28,28,1], "data2":[1,28,28,1]}

    • If using the CLI, {\"data1\": [1,28,28,1], \"data2\":[1,28,28,1]}

• KERAS: You must specify the name and shape (NCHW format) of expected data inputs using a dictionary format for your trained model. Note that while Keras model artifacts should be uploaded in NHWC (channel-last) format, DataInputConfig should be specified in NCHW (channel-first) format. The dictionary formats required for the console and CLI are different.

  • Examples for one input:

    • If using the console, {"input_1":[1,3,224,224]}

    • If using the CLI, {\"input_1\":[1,3,224,224]}

  • Examples for two inputs:

    • If using the console, {"input_1": [1,3,224,224], "input_2":[1,3,224,224]}

    • If using the CLI, {\"input_1\": [1,3,224,224], \"input_2\":[1,3,224,224]}

• MXNET/ONNX: You must specify the name and shape (NCHW format) of the expected data inputs in order using a dictionary format for your trained model. The dictionary formats required for the console and CLI are different.

  • Examples for one input:

    • If using the console, {"data":[1,3,1024,1024]}

    • If using the CLI, {\"data\":[1,3,1024,1024]}

  • Examples for two inputs:

    • If using the console, {"var1": [1,1,28,28], "var2":[1,1,28,28]}

    • If using the CLI, {\"var1\": [1,1,28,28], \"var2\":[1,1,28,28]}

• PyTorch: You can either specify the name and shape (NCHW format) of expected data inputs in order using a dictionary format for your trained model or you can specify the shape only using a list format. The dictionary formats required for the console and CLI are different. The list formats for the console and CLI are the same.

  • Examples for one input in dictionary format:

    • If using the console, {"input0":[1,3,224,224]}

    • If using the CLI, {\"input0\":[1,3,224,224]}

  • Example for one input in list format: [[1,3,224,224]]

  • Examples for two inputs in dictionary format:

    • If using the console, {"input0":[1,3,224,224], "input1":[1,3,224,224]}

    • If using the CLI, {\"input0\":[1,3,224,224], \"input1\":[1,3,224,224]}

  • Example for two inputs in list format: [[1,3,224,224], [1,3,224,224]]

• XGBOOST: input data name and shape are not needed.

framework

public Framework framework()

Identifies the framework in which the model was trained. For example: TENSORFLOW.

If the service returns an enum value that is not available in the current SDK version, framework will return Framework.UNKNOWN_TO_SDK_VERSION. The raw value returned by the service is available from frameworkAsString().

Returns:

Identifies the framework in which the model was trained. For example: TENSORFLOW.

See Also:

Framework

frameworkAsString

public String frameworkAsString()

Identifies the framework in which the model was trained. For example: TENSORFLOW.

If the service returns an enum value that is not available in the current SDK version, framework will return Framework.UNKNOWN_TO_SDK_VERSION. The raw value returned by the service is available from frameworkAsString().

Returns:

Identifies the framework in which the model was trained. For example: TENSORFLOW.

See Also:

Framework

toBuilder

public InputConfig.Builder toBuilder()

Specified by:

toBuilder in interface ToCopyableBuilder<InputConfig.Builder,InputConfig>

builder

public static InputConfig.Builder builder()

serializableBuilderClass

public static Class<? extends InputConfig.Builder> serializableBuilderClass()

hashCode

public int hashCode()

Overrides:

hashCode in class Object

equals

public boolean equals(Object obj)

Overrides:

equals in class Object

equalsBySdkFields

public boolean equalsBySdkFields(Object obj)

Specified by:

equalsBySdkFields in interface SdkPojo

toString

public String toString()

Returns a string representation of this object. This is useful for testing and debugging. Sensitive data will be redacted from this string using a placeholder value.

Overrides:

toString in class Object

getValueForField

public <T> Optional<T> getValueForField(String fieldName,
                                        Class<T> clazz)

sdkFields

public List<SdkField<?>> sdkFields()

Specified by:

sdkFields in interface SdkPojo