machine-learning - 按提交 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 设置为标签，然后搜索它，您可以在以下链接中查看如何设置标签

Answer 2

在最新版本 (1.21.0) 中，您可以使用此搜索字符串按提交哈希进行过滤tags.mlflow.source.git.commit = "XY"。可能在您最初发布该问题后引入了此功能。