CreateInferenceExperiment
Creates an inference experiment using the configurations specified in the request.
Use this API to setup and schedule an experiment to compare model variants on a Amazon SageMaker inference endpoint. For more information about inference experiments, see Shadow tests.
Amazon SageMaker begins your experiment at the scheduled time and routes traffic to your endpoint's model variants based on your specified configuration.
While the experiment is in progress or after it has concluded, you can view metrics that compare your model variants. For more information, see View, monitor, and edit shadow tests.
Request Syntax
{
"DataStorageConfig": {
"ContentType": {
"CsvContentTypes": [ "string
" ],
"JsonContentTypes": [ "string
" ]
},
"Destination": "string
",
"KmsKey": "string
"
},
"Description": "string
",
"EndpointName": "string
",
"KmsKey": "string
",
"ModelVariants": [
{
"InfrastructureConfig": {
"InfrastructureType": "string
",
"RealTimeInferenceConfig": {
"InstanceCount": number
,
"InstanceType": "string
"
}
},
"ModelName": "string
",
"VariantName": "string
"
}
],
"Name": "string
",
"RoleArn": "string
",
"Schedule": {
"EndTime": number
,
"StartTime": number
},
"ShadowModeConfig": {
"ShadowModelVariants": [
{
"SamplingPercentage": number
,
"ShadowModelVariantName": "string
"
}
],
"SourceModelVariantName": "string
"
},
"Tags": [
{
"Key": "string
",
"Value": "string
"
}
],
"Type": "string
"
}
Request Parameters
For information about the parameters that are common to all actions, see Common Parameters.
The request accepts the following data in JSON format.
- DataStorageConfig
-
The Amazon S3 location and configuration for storing inference request and response data.
This is an optional parameter that you can use for data capture. For more information, see Capture data.
Type: InferenceExperimentDataStorageConfig object
Required: No
- Description
-
A description for the inference experiment.
Type: String
Length Constraints: Maximum length of 1024.
Pattern:
.*
Required: No
- EndpointName
-
The name of the Amazon SageMaker endpoint on which you want to run the inference experiment.
Type: String
Length Constraints: Maximum length of 63.
Pattern:
^[a-zA-Z0-9](-*[a-zA-Z0-9]){0,62}
Required: Yes
- KmsKey
-
The Amazon Key Management Service (Amazon KMS) key that Amazon SageMaker uses to encrypt data on the storage volume attached to the ML compute instance that hosts the endpoint. The
KmsKey
can be any of the following formats:-
KMS key ID
"1234abcd-12ab-34cd-56ef-1234567890ab"
-
Amazon Resource Name (ARN) of a KMS key
"arn:aws:kms:us-west-2:111122223333:key/1234abcd-12ab-34cd-56ef-1234567890ab"
-
KMS key Alias
"alias/ExampleAlias"
-
Amazon Resource Name (ARN) of a KMS key Alias
"arn:aws:kms:us-west-2:111122223333:alias/ExampleAlias"
If you use a KMS key ID or an alias of your KMS key, the Amazon SageMaker execution role must include permissions to call
kms:Encrypt
. If you don't provide a KMS key ID, Amazon SageMaker uses the default KMS key for Amazon S3 for your role's account. Amazon SageMaker uses server-side encryption with KMS managed keys forOutputDataConfig
. If you use a bucket policy with ans3:PutObject
permission that only allows objects with server-side encryption, set the condition key ofs3:x-amz-server-side-encryption
to"aws:kms"
. For more information, see KMS managed Encryption Keys in the Amazon Simple Storage Service Developer Guide.The KMS key policy must grant permission to the IAM role that you specify in your
CreateEndpoint
andUpdateEndpoint
requests. For more information, see Using Key Policies in Amazon KMS in the Amazon Key Management Service Developer Guide.Type: String
Length Constraints: Maximum length of 2048.
Pattern:
^[a-zA-Z0-9:/_-]*$
Required: No
-
- ModelVariants
-
An array of
ModelVariantConfig
objects. There is one for each variant in the inference experiment. EachModelVariantConfig
object in the array describes the infrastructure configuration for the corresponding variant.Type: Array of ModelVariantConfig objects
Array Members: Minimum number of 1 item. Maximum number of 2 items.
Required: Yes
- Name
-
The name for the inference experiment.
Type: String
Length Constraints: Minimum length of 1. Maximum length of 120.
Pattern:
^[a-zA-Z0-9](-*[a-zA-Z0-9]){0,119}
Required: Yes
- RoleArn
-
The ARN of the IAM role that Amazon SageMaker can assume to access model artifacts and container images, and manage Amazon SageMaker Inference endpoints for model deployment.
Type: String
Length Constraints: Minimum length of 20. Maximum length of 2048.
Pattern:
^arn:aws[a-z\-]*:iam::\d{12}:role/?[a-zA-Z_0-9+=,.@\-_/]+$
Required: Yes
- Schedule
-
The duration for which you want the inference experiment to run. If you don't specify this field, the experiment automatically starts immediately upon creation and concludes after 7 days.
Type: InferenceExperimentSchedule object
Required: No
- ShadowModeConfig
-
The configuration of
ShadowMode
inference experiment type. Use this field to specify a production variant which takes all the inference requests, and a shadow variant to which Amazon SageMaker replicates a percentage of the inference requests. For the shadow variant also specify the percentage of requests that Amazon SageMaker replicates.Type: ShadowModeConfig object
Required: Yes
- Tags
-
Array of key-value pairs. You can use tags to categorize your Amazon resources in different ways, for example, by purpose, owner, or environment. For more information, see Tagging your Amazon Resources.
Type: Array of Tag objects
Array Members: Minimum number of 0 items. Maximum number of 50 items.
Required: No
- Type
-
The type of the inference experiment that you want to run. The following types of experiments are possible:
-
ShadowMode
: You can use this type to validate a shadow variant. For more information, see Shadow tests.
Type: String
Valid Values:
ShadowMode
Required: Yes
-
Response Syntax
{
"InferenceExperimentArn": "string"
}
Response Elements
If the action is successful, the service sends back an HTTP 200 response.
The following data is returned in JSON format by the service.
- InferenceExperimentArn
-
The ARN for your inference experiment.
Type: String
Length Constraints: Maximum length of 256.
Pattern:
arn:aws[a-z\-]*:sagemaker:[a-z0-9\-]*:[0-9]{12}:inference-experiment/.*
Errors
For information about the errors that are common to all actions, see Common Errors.
- ResourceInUse
-
Resource being accessed is in use.
HTTP Status Code: 400
- ResourceLimitExceeded
-
You have exceeded an SageMaker resource limit. For example, you might have too many training jobs created.
HTTP Status Code: 400
See Also
For more information about using this API in one of the language-specific Amazon SDKs, see the following: