如何在我的 python 笔记本中打印出 Sagemaker 批量转换作业状态？

Question

我是运行一个 python 笔记本，它正在 Sagemaker 中启动批量转换作业。但是，我还想在作业完成后打印状态“失败”、“进行中”和“完成”运行。截至目前，我只能启动批量转换作业（rf=随机森林），但我不确定如何打印作业状态。有人可以根据下面的脚本提供帮助吗？

rf_transformer = rf.transformer(
                                instance_count,
                                instance_type,
                                strategy=strategy,
                                output_path=output_path,
                                max_payload=max_payload)

rf_transformer.transform(
                                str('s3://batch_scoring/rf_output),
                                content_type='text/csv',
                                compression_type='Gzip'
                         )

Answer 1

您可以使用：

job_name = rf_transformer.latest_transform_job.name
rf_transformer.sagemaker_session.describe_transform_job(job_name)['TransformJobStatus']

您也可以直接使用 AWS SDK，如果您愿意：

import boto3

sagemaker_client = boto3.client('sagemaker')
sagemaker_client.describe_transform_job(job_name)['TransformJobStatus']

API 文档：https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_DescribeTransformJob.html

如何在我的 python 笔记本中打印出 Sagemaker 批量转换作业状态？

How to get Sagemaker Batch Transform Job status printed out in my python notebook?

amazon-s3

python-3.x

amazon-sagemaker