pipelines/components/aws/sagemaker/ModelExplainabilityJobDefin.../component.yaml

125 lines
3.9 KiB
YAML

name: "Sagemaker - ModelExplainabilityJobDefinition"
description: Create ModelExplainabilityJobDefinition
inputs:
- {
name: region,
type: String,
description: "The region to use for the training job",
}
###########################GENERATED SECTION BELOW############################
- {
name: job_definition_name,
type: String,
default: '',
description: "The name of the model explainability job definition.",
}
- {
name: job_resources,
type: JsonObject,
default: '{}',
description: "Identifies the resources to deploy for a monitoring job.",
}
- {
name: model_explainability_app_specification,
type: JsonObject,
default: '{}',
description: "Configures the model explainability job to run a specified Docker container image.",
}
- {
name: model_explainability_baseline_config,
type: JsonObject,
default: '{}',
description: "The baseline configuration for a model explainability job.",
}
- {
name: model_explainability_job_input,
type: JsonObject,
default: '{}',
description: "Inputs for the model explainability job.",
}
- {
name: model_explainability_job_output_config,
type: JsonObject,
default: '{}',
description: "The output configuration for monitoring jobs.",
}
- {
name: network_config,
type: JsonObject,
default: '{}',
description: "Networking options for a model explainability job.",
}
- {
name: role_arn,
type: String,
default: '',
description: "The Amazon Resource Name (ARN) of an IAM role that Amazon SageMaker can assume to perform tasks on your behalf.",
}
- {
name: stopping_condition,
type: JsonObject,
default: '{}',
description: "A time limit for how long the monitoring job is allowed to run before stopping.",
}
- {
name: tags,
type: JsonArray,
default: '[]',
description: "(Optional) An array of key-value pairs.",
}
###########################GENERATED SECTION ABOVE############################
outputs:
###########################GENERATED SECTION BELOW############################
- {
name: ack_resource_metadata,
type: JsonObject,
description: "All CRs managed by ACK have a common `Status.",
}
- {
name: conditions,
type: JsonArray,
description: "All CRS managed by ACK have a common `Status.",
}
- {
name: sagemaker_resource_name,
type: String,
description: "Resource name on Sagemaker",
}
###########################GENERATED SECTION ABOVE############################
implementation:
container:
image: public.ecr.aws/kubeflow-on-aws/aws-sagemaker-kfp-components:2.3.0
command: [python3]
args:
- ModelExplainabilityJobDefinition/src/ModelExplainabilityJobDefinition_component.py
- --region
- { inputValue: region }
###########################GENERATED SECTION BELOW############################
- --job_definition_name
- { inputValue: job_definition_name }
- --job_resources
- { inputValue: job_resources }
- --model_explainability_app_specification
- { inputValue: model_explainability_app_specification }
- --model_explainability_baseline_config
- { inputValue: model_explainability_baseline_config }
- --model_explainability_job_input
- { inputValue: model_explainability_job_input }
- --model_explainability_job_output_config
- { inputValue: model_explainability_job_output_config }
- --network_config
- { inputValue: network_config }
- --role_arn
- { inputValue: role_arn }
- --stopping_condition
- { inputValue: stopping_condition }
- --tags
- { inputValue: tags }
###########################GENERATED SECTION ABOVE############################