按提交 ID 过滤 mlflow 运行

Question

使用 MlFlow 的 UI 时，是否可以使用 (git) 提交 ID 过滤/搜索运行？我设法按参数搜索，但似乎没有办法按提交 ID 进行过滤。

Answer 1

ui 中的搜索将转换为对 mlflow 跟踪服务器的 REST API 调用： https://mlflow.org/docs/latest/rest-api.html#search-runs

您可以看到，您可以搜索超过 3 个内容：参数、指标和标签。

您可以在训练模型时将 git commit id 设置为标签，然后搜索它，您可以在以下链接中查看如何设置标签