MLflow：一个管理机器学习生命周期的工具

MLflow 是一个开源平台，专门用于帮助机器学习从业者和团队处理机器学习过程的复杂性。MLflow 专注于机器学习项目的完整生命周期，确保每个阶段都可管理、可追溯和可复现。

MLflow 入门资源

如果您是第一次探索 MLflow，这里的教程和指南是一个很好的起点。这些教程的重点是让您尽快熟悉基本功能、术语、API 和使用 MLflow 的一般最佳实践，以增强您在特定领域指南和教程中的学习。

了解 MLflow

了解 MLflow 的核心组件

快速入门

通过我们的5 分钟教程开始使用 MLflow

指南

通过本深入跟踪指南学习 MLflow 的核心组件

MLflow 基础

了解如何在 MLflow 中执行常见任务

指南

自动日志记录指南，实现轻松模型跟踪

模型签名和 MLflow 中的类型验证

模型部署快速入门

使用 MLflow 进行超参数调整

MLflow 模型简介

了解 MLflow 模型相关主题

指南

自定义 Python 模型简介

MLflow 中的模型依赖管理

模型签名和类型验证

传统机器学习

开始使用 MLflow 的传统机器学习集成

快速入门和指南

MLflow Scikit-learn 简介

MLflow XGBoost 使用方法

Spark ML 指南

MLflow Prophet 时间序列预测

深度学习

开始使用 MLflow 的深度学习库集成

指南和教程

TensorFlow

PyTorch

Keras

用于 NLP 工作流的spaCy

使用 MLflow 进行传统机器学习和深度学习

MLflow 为传统机器学习和深度学习工作流提供全面支持。从实验跟踪和模型版本控制到部署和监控，MLflow 简化了 ML 生命周期中的每个方面。无论您是使用 scikit-learn 模型、训练深度神经网络还是管理复杂的 ML 管道，MLflow 都提供了构建可靠、可扩展的机器学习系统所需的工具。

探索下面的核心 MLflow 功能和集成，以增强您的 ML 开发工作流！

跟踪与实验

跟踪实验并管理您的 ML 开发

核心功能

MLflow Tracking 提供全面的实验日志记录、参数跟踪、指标可视化和工件管理。

主要优势

• 实验组织：跟踪和比较多个模型实验
• 指标可视化：用于模型性能的内置图表
• 工件存储：存储模型、图表和其他文件，每次运行一个
• 协作：跨团队共享实验和结果

指南

跟踪入门

高级跟踪功能

流行库的自动日志记录

模型注册表

管理模型版本和生命周期

核心功能

MLflow Model Registry 提供集中式模型版本控制、阶段管理和模型血缘跟踪。

主要优势

• 版本控制：使用自动血缘跟踪模型版本
• 阶段管理：在暂存、生产和归档阶段推广模型
• 协作：基于团队的模型审查和批准工作流
• 模型发现：在组织内搜索和发现模型

指南

模型注册表简介

注册您的第一个模型

模型部署

将模型部署到生产环境

核心功能

MLflow 部署支持多个部署目标，包括 REST API、云平台和边缘设备。

主要优势

• 多个目标：部署到本地服务器、云平台或容器化环境
• 模型服务：内置 REST API 服务，带自动输入验证
• 批量推理：支持批量评分和离线预测
• 生产就绪：适用于企业用途的可扩展部署选项

指南

模型部署概述

本地模型服务

云部署选项

Kubernetes 部署

机器学习库集成

探索原生 MLflow 机器学习库集成

Scikit-learn

XGBoost

TensorFlow

PyTorch

Keras

Spark MLlib

模型评估

评估和验证您的 ML 模型

核心功能

MLflow 评估提供全面的模型验证工具、自动化指标计算和模型比较功能。

主要优势

• 自动化指标：内置的分类、回归和模型评估指标
• 自定义评估器：为特定领域指标创建自定义评估函数
• 模型比较：并排比较多个模型和版本
• 验证数据集：跟踪评估数据集并确保可复现结果

指南

了解如何使用 MLflow 评估您的 ML 模型

发现自定义评估指标和函数

使用MLflow 模型比较来比较模型

随处运行 MLflow

MLflow 可以在各种环境中使用，包括您的本地环境、本地集群、云平台和托管服务。作为开源平台，MLflow 是供应商中立的；无论您在哪里进行机器学习，您都可以访问 MLflow 的核心功能集，例如跟踪、评估、可观察性等等。