例: ハイパーパラメータ調整ジョブ

この例では、ハイパーパラメータ調整ジョブを設定して開始するための新しいノートブックを作成する方法を示します。調整ジョブは、Amazon SageMaker AI を使用した XGBoost アルゴリズム を使用してモデルをトレーニングし、顧客が電話で連絡を受けた後に銀行で定期預金を申し込むかどうかを予測します。

低レベル SDK for Python (Boto3) を使用してハイパーパラメータ調整ジョブを設定して起動し、 AWS Management Console を使用してハイパーパラメータ調整ジョブのステータスをモニタリングします。Amazon SageMaker AI の高レベル Amazon SageMaker Python SDK を使用して、ハイパーパラメータ調整ジョブを設定、実行、モニタリング、分析することもできます。詳しくは、https://github.com/aws/sagemaker-python-sdk を参照してください。

前提条件

この例のコードを実行するには、以下が必要です。

• AWS アカウントと管理者ユーザー

• トレーニングデータセットとトレーニング中に作成されたモデルアーティファクトを保存するための Amazon S3 バケット

• 実行中の SageMaker AI ノートブックインスタンス

トピック

• ノートブックインスタンスの作成

• Amazon SageMaker AI Boto 3 クライアントを取得する

• SageMaker AI 実行ロールを取得する

• Amazon S3 バケットを入力と出力に使用する

• トレーニングデータをダウンロード、準備、アップロードする

• ハイパーパラメータ調整ジョブを設定して開始する

• クリーンアップ