aws / aws/sagemaker-python-sdk
MLFlow E2E Example Notebook
- 主要语言
- Python
- 星标
- 2.3k
- 派生
- 1.3k
- 平均合并
- 1 天 22 小时
- 30 天内合并 PR
- 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 tracking server ARN 前置条件。创建 v3-mlflow-train-inference-e2e-example.ipynb,涵盖使用 MLflow logging 的 PyTorch 训练、模型注册、通过 ModelBuilder 部署以及端点测试;完整工作流得到演示即表示完成。
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- jupyter-notebook, python, pytorch
- 领域
- cloud, documentation, machine-learning
- Issue 类型
- 功能
- 难度
- 3/5
- 预计耗时
- 1-2 天
- 活跃度
- 停滞
- 描述清晰度
- 描述清楚
- 新手友好度
- 25/100