我使用SageMaker Python SDK创建了一个超参数调优任务,该任务并行运行多个任务以寻找最佳的超参数值。
任务完成后,我得到了最佳训练任务的名称,形式为字符串“Job…”。我找到了一篇文章,介绍了如何使用AWS-CLI或HTTP请求来描述一个任务。
是否有使用Python SageMaker SDK的方法来做到这一点,以避免向AWS发起认证请求的复杂性?
回答:
通过使用sagemaker.session.Session
实例,您可以描述训练任务:
import sagemakersagemaker_session = sagemaker.session.Session()sagemaker_session.describe_training_job("Job...")