@@ -362,6 +362,52 @@ Example:
362
362
prefix = model_prefix
363
363
)
364
364
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
+
365
411
CreateModelStep
366
412
````````````````
367
413
Referable Property List:
0 commit comments