创建 Amazon SageMaker 实验

导入sys模块来安装软件开发工具包。

import sys

（可选）Amazon SageMaker Python 开发工具包, 预安装在 SageMaker 工作室. 如果您计划在工作室外运行代码，请安装 SageMaker Python 软件开发工具包。

!{sys.executable} -m pip install sagemaker

安装SageMaker 实验 Python 开发工具包.

!{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': 'demo-experiments', 'Value': 'demo1'}])

（工作室）要查看 SageMaker 工作室中的实验，请在左侧栏中选择组成部分和登记册图标 ( )。在下拉菜单中，选择实验和试验以显示实验浏览器。

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

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

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

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

创建试验组件，将其在试验中创建。试用组件是 SageMaker 作业。

添加实验配置参数设置为相应的方法。支持下表中列出的 SageMaker 作业。

以下示例适用于训练作业。这些区域有：Tags参数将标签添加到试用组件中。ExperimentName，因为在早期步骤中创建试验时，试验与实验相关联。

使用 SageMaker Python 开发工具包

sagemaker.estimator.Estimator(
    ...,
    sagemaker_session = sm_sess,
    tags = [{'Key': 'demo-jobs', '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': 'demo-jobs', 'Value': 'demo2'}])

(Studio) 在试用列表中，双击试验以显示试验中组件的列表（此示例有一个试验）。您的列表应类似于以下内容：

(Studio) 要查看有关实验、试验和作业（试验组件）的信息，请参阅查看和比较 Amazon SageMaker 实验、试验和试验组件.