@@ -362,6 +362,52 @@ Example:
362362 prefix = model_prefix
363363 )
364364
365+ AutoMLStep
366+ `````````````
367+ Referable Property List:
368+
369+ - `DescribeAutoMLJob `_
370+ - `BestCandidateProperties.ModelInsightsJsonReportPath `_
371+ - `BestCandidateProperties.ExplainabilityJsonReportPath `_
372+
373+ .. _DescribeAutoMLJob : https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeAutoMLJob
374+ .. _BestCandidateProperties.ModelInsightsJsonReportPath : https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CandidateArtifactLocations.html#sagemaker-Type-CandidateArtifactLocations-ModelInsights
375+ .. _BestCandidateProperties.ExplainabilityJsonReportPath : https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CandidateArtifactLocations.html#sagemaker-Type-CandidateArtifactLocations-Explainability
376+
377+ Example:
378+
379+ .. code-block :: python
380+
381+ step_automl = AutoMLStep(... )
382+
383+ auto_ml_model = step_automl.get_best_model(< role> )
384+
385+ model_metrics = ModelMetrics(
386+ model_statistics = MetricsSource(
387+ s3_uri = auto_ml_step.properties.BestCandidateProperties.ModelInsightsJsonReportPath,
388+ content_type = " application/json" ,
389+ ),
390+ explainability = MetricsSource(
391+ s3_uri = auto_ml_step.properties.BestCandidateProperties.ExplainabilityJsonReportPath,
392+ content_type = " application/json" ,
393+ )
394+ )
395+
396+ step_args_register_model = auto_ml_model.register(
397+ content_types = [" text/csv" ],
398+ response_types = [" text/csv" ],
399+ inference_instances = [" ml.t2.medium" , " ml.m5.large" ],
400+ transform_instances = [" ml.m5.large" ],
401+ model_package_group_name = " My model package group name" ,
402+ approval_status = " PendingManualApproval" ,
403+ model_metrics = model_metrics,
404+ )
405+
406+ step_register_model = ModelStep(
407+ name = " auto-ml-model-register" ,
408+ step_args = step_args_register_model,
409+ )
410+
365411 CreateModelStep
366412````````````````
367413Referable Property List:
0 commit comments