kedro-mlflow

在 Kedro 管道中一键启用 MLflow 实验追踪与模型版本管理

kedro-mlflow 是一个 Kedro 插件，旨在将 MLflow 的机器学习生命周期管理能力无缝集成到 Kedro 数据管道项目中。它解决了 Kedro 项目在模型版本管理、实验追踪和模型打包方面的痛点，使得数据科学家和工程师可以在不离开 Kedro 工作流的前提下，利用 MLflow 强大的实验记录、模型注册和部署功能。该插件通过自动配置 MLflow 跟踪 URI、提供 Kedro 节点级别的参数和指标记录、以及一键打包模型为 MLflow 格式，简化了从数据准备到模型上线的整个流程。核心能力包括：自动实验追踪、模型版本控制、与 Kedro 数据集和参数系统的深度集成，以及通过 CLI 命令快速启动 MLflow UI 和打包模型。