跳到主要内容

MLflow 追踪 UI

MLflow 实验中的追踪​

记录追踪后，你可以在 MLflow UI 中查看它们，位于主实验页面的“追踪”选项卡下。如果你的追踪是在运行上下文中记录的，此选项卡在单个运行页面中也可用。

MLflow Tracking UI

此表包含有关追踪的高级信息，例如追踪 ID、根 Span 的输入/输出等。在此页面上，你还可以执行一些操作来管理你的追踪

使用 UI 中的搜索栏，你可以根据名称、标签或其他元数据轻松筛选追踪。有关查询字符串格式的详细信息，请查看搜索文档。

Searching traces

浏览 Span 数据​

要浏览单个追踪的 Span 数据，只需点击“追踪 ID”或“追踪名称”列中的链接即可打开追踪查看器

Trace Browser

Jupyter Notebook 集成​

注意

MLflow 追踪 Jupyter 集成功能在 MLflow 2.20 及更高版本中可用。

你也可以在 Jupyter Notebook 中直接查看追踪 UI，从而无需离开开发环境即可调试应用程序。

Jupyter Trace UI

此功能需要使用 MLflow 追踪服务器，因为 UI 资产是从这里获取的。要开始使用，只需确保 MLflow 追踪 URI 已设置为你的追踪服务器（例如 mlflow.set_tracking_uri("https://:5000")）。

默认情况下，追踪 UI 将在以下事件中自动显示

  1. 当单元格代码生成追踪时（例如通过自动追踪，或通过运行手动追踪的函数）
  2. 当调用mlflow.search_traces()时
  3. 当显示 mlflow.entities.Trace() 对象时（例如通过 IPython 的 display 函数，或当它是单元格中返回的最后一个值时）

要禁用显示，只需调用 mlflow.tracing.disable_notebook_display()，然后重新运行包含 UI 的单元格。要再次启用它，请调用 mlflow.tracing.enable_notebook_display()。