aws / aws/sagemaker-python-sdk

MLFlow E2E Example Notebook

オープン
#5,513 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る
主要言語
Python
スター
2.3k
フォーク
1.3k
平均マージ
1日 22時間
マージ済み PR(30日)
35

説明

**PR**: https://github.com/aws/sagemaker-python-sdk/pull/5514

**Describe the feature you'd like**

Add a new example notebook demonstrating the complete end-to-end workflow from training a PyTorch model to deploying it for inference on SageMaker, with MLflow 3.x tracking and model registry integration.

**How would this feature be used? Please describe.**

Users who want to leverage MLflow with SageMaker V3 SDK currently lack a comprehensive example showing the full workflow. This notebook would demonstrate:

1. Connecting to SageMaker MLflow tracking server
2. Training a PyTorch model with ModelTrainer while logging metrics/params to MLflow
3. Registering the trained model to MLflow Model Registry
4. Deploying directly from MLflow registry using ModelBuilder
5. Testing the deployed endpoint

Example workflow:

```
# Train with MLflow logging
model_trainer = ModelTrainer(training_image=..., source_code=...)
model_trainer.train()

# Deploy from MLflow registry
model_builder = ModelBuilder(
model_metadata={"MLFLOW_MODEL_PATH": "models:/my-model/1", ...}
)
model_builder.build()
model_builder.deploy()

```
**Describe alternatives you've considered**

Existing notebooks cover training or inference separately, but none show the integrated MLflow workflow end-to-end.

**Additional context**

Target location: v3-examples/ml-ops-examples/
Notebook name: v3-mlflow-train-inference-e2e-example.ipynb
Prerequisites: SageMaker MLflow App (tracking server ARN)

コントリビューションガイド

コントリビューションガイドを開く

調査の方向性

v3-examples/ml-ops-examples/ から始め、SageMaker MLflow App のトラッキングサーバー ARN の前提条件を使用します。MLflow ロギングを伴う PyTorch トレーニング、モデル登録、ModelBuilder によるデプロイ、エンドポイントテストを網羅する v3-mlflow-train-inference-e2e-example.ipynb を作成します。完全なワークフローが実証されれば完了です。

索引モデルが issue の本文から書いたものです。

評価

技術スタック
jupyter-notebook, python, pytorch
領域
cloud, documentation, machine-learning
issue の種類
機能追加
難易度
3/5
見積もり時間
1〜2日
活発さ
停滞
明瞭さ
明確に書かれている
初心者へのやさしさ
25/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。