跳到主要内容

用于传统机器学习的 MLflow

MLflow for ML Models 为传统机器学习工作流提供了全面的实验跟踪、模型管理和部署功能。从 scikit-learn 流水线到梯度提升模型,MLflow 简化了从实验到生产的路径。

快速开始

为什么要为传统机器学习选择 MLflow?

自动日志记录

仅需一行代码 (mlflow.autolog()) 即可为 scikit-learn、XGBoost、LightGBM 等捕获参数、指标、模型和工件。

实验组织

通过父子运行来跟踪超参数搜索。通过可视化图表和可排序表格跨算法比较模型。

流水线跟踪

自动记录 scikit-learn 流水线组件、预处理步骤和特征转换,实现完全的可复现性。

灵活部署

通过 Docker、Kubernetes 和云平台支持,部署用于实时推理、批处理或边缘计算的模型。

支持的库

scikit learn
XGBoost Logo
Spark Logo
LightGBM Logo
CatBoost Logo
Statsmodels Logo
Prophet Logo

了解更多