你当前正在访问 Microsoft Azure Global Edition 技术文档网站。如果需要访问由世纪互联运营的 Microsoft Azure 中国技术文档网站，请访问 https://docs.azure.cn。

使用 MLflow 跟踪试验和模型

2025-11-01

跟踪是保存试验相关信息的过程。本文介绍如何使用 MLflow 跟踪 Azure 机器学习工作区中的试验和运行。

使用 Azure 机器学习时，MLflow API 中提供的某些方法可能不可用。有关支持和不支持的操作的详细信息，请参阅查询运行和试验的支持矩阵。还可以从 MLflow 和 Azure 机器学习一文中了解 Azure 机器学习中支持的 MLflow 功能。

注意

若要跟踪在 Azure Databricks 上运行的试验，请参阅使用 MLflow 和 Azure 机器学习跟踪 Azure Databricks 机器学习试验。
若要跟踪在 Azure Synapse Analytics 上运行的试验，请参阅使用 MLflow 和 Azure 机器学习跟踪 Azure Synapse Analytics ML 试验。

Prerequisites

拥有 Azure 订阅和 Azure 机器学习的免费或付费版本。
若要运行 Azure CLI 和 Python 命令，请安装 Azure CLI v2 和适用于 Python 的 Azure 机器学习 SDK v2。首次运行 Azure 机器学习 CLI 命令时，会自动安装 Azure CLI 的 ml 扩展。

安装 MLflow SDK mlflow 包和适用于 MLflow 的 Azure 机器学习 azureml-mlflow 插件：
```
pip install mlflow azureml-mlflow
```
提示

可以使用 mlflow-skinny 包，它是一个不带 SQL 存储、服务器、UI 或数据科学依赖项的轻型 MLflow 包。对于主要需要 MLflow 跟踪和日志记录功能但不是完整的功能套件（包括部署）的用户，我们建议使用此包。
创建 Azure 机器学习工作区。若要创建工作区，请参阅创建入门所需的资源。查看在工作区中执行 MLflow 操作所需的访问权限。
若要执行远程跟踪（即跟踪在 Azure 机器学习以外运行的试验），请将 MLflow 配置为指向 Azure 机器学习工作区的跟踪 URI。有关如何将 MLflow 连接到工作区的详细信息，请参阅为 Azure 机器学习配置 MLflow。

配置试验

MLflow 组织试验和运行中的信息。在 Azure 机器学习中，运行称为“作业”。默认情况下，运行记录到自动创建的名为“默认”的试验，但你可配置要跟踪的试验。

Notebook
作业

对于交互式训练（如 Jupyter 笔记本中），请使用 MLflow 命令 mlflow.set_experiment()。例如，以下代码片段会配置一个试验：

experiment_name = 'hello-world-example'
mlflow.set_experiment(experiment_name)

若要使用 Azure 机器学习 CLI 或 SDK 提交作业，请使用作业的 experiment_name 属性设置试验名称。无需在训练脚本中配置试验名称。

$schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
command: echo "hello world"
environment:
  image: library/python:latest
tags:
  hello: world
display_name: hello-world-example
experiment_name: hello-world-example
description: |

配置运行

Azure 机器学习会跟踪 MLflow 调用运行的训练作业。使用运行捕获作业执行的所有处理。

Notebook
作业

以交互方式工作时，一旦你记录需要活动运行的信息，MLflow 就会立即开始跟踪训练例程。例如，如果启用了 MLflow 的自动记录功能，则记录指标或参数或启动训练周期时，MLflow 跟踪将启动。

但显式启动运行通常很有帮助，特别是在要捕获“持续时间”字段中试验总时间的情况下。要显式启动运行，请使用 mlflow.start_run()。

无论是否手动启动运行，你最终都需要停止运行，以便 MLflow 知道试验运行已完成，并且可以将运行状态标记为“已完成”。若要停止运行，请使用 mlflow.end_run()。

以下代码会手动启动运行，并在笔记本末尾结束该运行：

mlflow.start_run()

# Your code

mlflow.end_run()

最好手动启动运行，以免忘记结束运行。可使用上下文管理器范式来帮助记住结束运行。

with mlflow.start_run() as run:
    # Your code

当您开始一个新的运行时使用 mlflow.start_run()，指定 run_name 参数非常有用，该参数随后会被转换为 Azure 机器学习用户界面中的运行名称。这样做有助于更快地识别运行。

with mlflow.start_run(run_name="hello-world-example") as run:
    # Your code

借助 Azure 机器学习作业，可以将长期训练或推理例程提交为独立且可重现的执行。

创建具有跟踪的训练例程

处理作业时，通常将所有训练逻辑作为文件包含在文件夹中，例如 src。其中一个文件是包含训练代码入口点的 Python 文件。

在训练例程中，可以使用 MLflow SDK 跟踪任何指标、参数、项目或模型。如需示例，请参阅使用 MLflow 记录指标、参数和文件。

以下示例显示了添加日志记录的 hello_world.py 训练例程：

# imports
import os
import mlflow

from random import random

# define functions
def main():
    mlflow.log_param("hello_param", "world")
    mlflow.log_metric("hello_metric", random())
    os.system(f"echo 'hello world' > helloworld.txt")
    mlflow.log_artifact("helloworld.txt")


# run functions
if __name__ == "__main__":
    # run main function
    main()

前面的代码示例不使用 mlflow.start_run()。如果使用此行，MLflow 将重复使用当前活动运行。因此，如果将代码迁移到 Azure 机器学习，则无需删除 mlflow.start_run() 。

确保作业的环境已安装 MLflow

所有 Azure 机器学习特选环境均已安装 MLflow。但是，如果使用自定义环境，请创建一个具有所需依赖项 的conda.yml 文件，并在作业中引用环境。

channels:
- conda-forge
dependencies:
- python=3.8.5
- pip
- pip:
  - mlflow
  - azureml-mlflow
  - fastparquet
  - cloudpickle==1.6.0
  - colorama==0.4.4
  - dask==2023.2.0

配置作业名称

使用 Azure 机器学习作业参数 display_name 配置运行的名称。

使用 display_name 属性配置作业。

Azure CLI
Python SDK

若要配置作业，请在 src 目录外部的 job.yml 文件中创建具有作业定义的 YAML 文件。

$schema: https://azuremlschemas.azureedge.net/latest/commandJob.schema.json
command: echo "hello world"
environment:
  image: library/python:latest
tags:
  hello: world
display_name: hello-world-example
experiment_name: hello-world-example
description: |

from azure.ai.ml import command, Environment

command_job = command(
    code="src",
    command="echo "hello world",
    environment=Environment(image="library/python:latest"),
    compute="cpu-cluster",
    display_name="hello-world-example"
)

请确保不要在训练例程中使用 mlflow.start_run(run_name="") 。

提交作业

工作区是 Azure 机器学习的顶级资源，提供一个集中位置来处理创建的所有 Azure 机器学习项目。连接到 Azure 机器学习工作区。

Azure CLI
Python SDK

az account set --subscription <subscription>
az configure --defaults workspace=<workspace> group=<resource-group> location=<location>

导入所需的库：

from azure.ai.ml import MLClient
from azure.identity import DefaultAzureCredential

配置工作区详细信息并获取工作区句柄：

subscription_id = "<subscription>"
resource_group = "<resource-group>"
workspace = "<workspace>"

ml_client = MLClient(DefaultAzureCredential(), subscription_id, resource_group, workspace)

打开终端并使用以下代码提交作业。使用 MLflow 并在 Azure 机器学习上运行的作业会自动将任何跟踪信息记录到工作区。
- Azure CLI
- Python SDK
使用 Azure 机器学习 CLI 提交作业。
```
az ml job create -f job.yml --web
```
使用 Python SDK 提交作业。
```
returned_job = ml_client.jobs.create_or_update(command_job)
returned_job.studio_url
```
在 Azure 机器学习工作室中监视作业进度。

启用 MLflow 自动日志记录

可以使用 MLflow 手动记录指标、参数和文件，还可依赖于 MLflow 的自动日志记录功能。 MLflow 支持的每个机器学习框架都会确定自动跟踪的内容。

若要启用自动日志记录，请在训练代码之前插入以下代码：

mlflow.autolog()

查看工作区中的指标和项目

你的工作区中将跟踪 MLflow 记录的指标和项目。可以在 Azure 机器学习工作室中查看和访问它们，也可以使用 MLflow SDK 以编程方式访问它们。

在工作室中查看指标和工件：

在工作区的“作业”页面上，选择试验名称。
在“试验详细信息”页面上，选择“指标”选项卡。
选择记录的指标以在页面右侧呈现图表。可以通过应用平滑处理、更改颜色或在单个图形上绘制多个指标来自定义图表。还可以调整布局的大小和重新排列布局。
创建所需的视图后，保存该视图以供将来使用，并使用直接链接与团队成员共享。

若要使用 MLflow SDK 以编程方式访问或查询指标、参数和项目，请使用mlflow.get_run（）。

import mlflow

run = mlflow.get_run("<RUN_ID>")

metrics = run.data.metrics
params = run.data.params
tags = run.data.tags

print(metrics, params, tags)

提示

上述示例仅返回给定指标的最后一个值。若要检索给定指标的所有值，请使用 mlflow.get_metric_history 方法。若要详细了解如何检索指标值，请参阅从运行中获取参数和指标。

若要下载已记录的项目（如文件和模型），请使用 mlflow.artifacts.download_artifacts()。

mlflow.artifacts.download_artifacts(run_id="<RUN_ID>", artifact_path="helloworld.txt")

有关如何使用 MLflow 在 Azure 机器学习中检索或比较试验和运行的信息的详细信息，请参阅查询和比较试验和运行与 MLflow。

反馈

此页面是否有帮助？