python - 如何更改 Databricks“运行”选项卡跟踪的实验？

Question

我正在尝试使用 mlflow databricks 集成，特别是跟踪 API。通常，我可以在笔记本的方便侧边栏中查看过去的运行信息，正如您在此处看到的以及我从教程中获得的那样。但是，我现在想要的是使用多个笔记本将运行发送到同一个实验。此外，我想查看每个笔记本中所有这些常见运行的结果。为此，我需要更改“运行”选项卡跟踪的（默认）实验。

最终，我的问题归结为以下几点：如何设置“运行”选项卡跟踪的实验？我试过使用mlflow.set_tracking_uri和mlflow.set_experiment(mlflow_experiment_name)

Answer 1

我不相信今天这是可能的，因为设计选择是将运行选项卡与笔记本实验相关联。从文档：

Databricks 工作区中的每个 Python 和 R 笔记本都有自己的实验。当您在笔记本中使用 MLflow 时，它会在笔记本实验中记录运行。

笔记本实验与其对应的笔记本共享相同的名称和 ID。笔记本 ID 是笔记本 URL 末尾的数字标识符。

您可以创建独立于笔记本实验的实验，并将来自不同来源的运行记录到其中。不过，您仍然需要打开跟踪 UI 来探索结果。

换句话说，您可以将来自不同笔记本的多次运行发送到同一个实验，但现在您无法将多次运行记录到特定笔记本的“运行”选项卡中。