跳到主要内容

MLflow 追踪 UI

MLflow 实验中的追踪

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

MLflow Tracking UI

此表包含有关追踪的高级信息,例如追踪 ID、根跨度的输入/输出等。您还可以从此页面执行一些操作来管理您的追踪。

使用 UI 中的搜索栏,您可以根据名称、标签或其他元数据轻松过滤您的追踪。有关查询字符串格式的详细信息,请参阅搜索文档

Searching traces

浏览跨度数据

要浏览单个追踪的跨度数据,只需点击“Trace ID”或“Trace name”列中的链接即可打开追踪查看器。

Trace Browser

Jupyter Notebook 集成

注意

MLflow Tracing Jupyter 集成在 MLflow 2.20 及更高版本中可用。

您也可以直接在 Jupyter Notebook 中查看追踪 UI,这样您就可以在不切换出开发环境的情况下调试您的应用程序。

Jupyter Trace UI

此功能需要使用 MLflow Tracking Server,因为 UI 资源是从此处获取的。要开始使用,只需确保 MLflow Tracking 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()