创建 Amazon SageMaker 实验

导入 sys 模块安装SDK。

import sys

(可选) Amazon SageMaker Python SDK,预装于 SageMaker 书房。如果您计划在Price外运行代码,请安装 SageMaker PythonSDK。

!{sys.executable} -m pip install sagemaker

安装 SageMaker 试验PythonSDK.

!{sys.executable} -m pip install sagemaker-experiments

导入模块。

import time
from time import strftime

import sagemaker

from smexperiments.experiment import Experiment
from smexperiments.trial import Trial
from smexperiments.trial_component import TrialComponent
from smexperiments.tracker import Tracker

获取执行角色并创建 SageMaker 教学课。

role = sagemaker.get_execution_role()
sm_sess = sagemaker.session.Session()

创建 SageMaker 实验。实验名称在您的帐户中必须是唯一的。

create_date = strftime("%Y-%m-%d-%H-%M-%S")
demo_experiment = Experiment.create(experiment_name = "DEMO-{}".format(create_date),
                                    description = "Demo experiment",
                                    tags = [{'Key': 'my-experiments', 'Value': 'demo1'}])

(研究)要查看 SageMaker 单间,在左侧边栏选择 SageMaker 实验列表 图标( )显示实验浏览器。

代码运行后,实验列表包含新的实验。列表可能需要一些时间刷新并显示实验。还会显示实验标签上的过滤器。仅显示具有匹配标记的实验。您的列表应类似于以下内容:

为实验创建试用版。试验名称在您的帐户中必须是唯一的。

demo_trial = Trial.create(trial_name = "DEMO-{}".format(create_date),
                          experiment_name = demo_experiment.experiment_name,
                          tags = [{'Key': 'my-experiments', 'Value': 'demo1'}])

(Studio)在实验列表中,双击实验以显示实验中的试验列表(此示例有一个试验)。您的列表应类似于以下内容:

创建试验组件作为试验的一部分。试验部分是 SageMaker 工作。

添加 实验配置 参数。的 SageMaker 支持下表中列出的作业。

以下示例适用于培训工作。的 Tags 参数为试验组件添加标记。ExperimentName 未指定,因为在早期步骤中创建试验时,试验与试验相关。

使用 SageMaker PythonSDK

sagemaker.estimator.Estimator(
    ...,
    sagemaker_session = sm_sess,
    tags = [{'Key': 'my-experiments', 'Value': 'demo2'}])

estimator.fit(
    ...,
    experiment_config = {
        # "ExperimentName"
        "TrialName" : demo_trial.trial_name,
        "TrialComponentDisplayName" : "TrainingJob",
    })

使用Boto3

create_training_job(
    ...,
    "ExperimentConfig": {
        # "ExperimentName"
        "TrialName" : demo_trial.trial_name,
        "TrialComponentDisplayName" : "TrainingJob",
    },
    "Tags": [{'Key': 'my-experiments', 'Value': 'demo2'}])

(研究)在试验列表中,双击试验以显示试验中的组件列表(此示例包含一项试验)。您的列表应类似于以下内容:

(研究)要查看有关实验、试验和工作(试验部分)的信息,请参阅 查看和比较 Amazon SageMaker 实验、试验和试验组成部分.