如何使用 MLflow 加载记录的字典？答案

【问题标题】：How to load a logged dictionary using MLflow?如何使用 MLflow 加载记录的字典？
【发布时间】：2022-12-09 08:53:44
【问题描述】：

我使用 MLflow 和以下代码成功地以字典格式记录了一个 pandas 数据帧：

df = <pd.DataFrame>
mlflow.log_dict(<pd.DataFrame>.to_dict(orient='dict'), f"my_logged_df_dict.json")

现在我试图阅读那个记录的字典并进一步调查它的内容。根据文档，这应该可以使用方便的mlflow.artifacts.load_dict() 方法轻松实现。但是，运行此命令后，我得到

AttributeError: module 'mlflow' has no attribute 'artifacts'

我检查了我的版本是 1.24.0。我在 Github 上检查了相应的 1.24.0 功能分支，我可以看到 artifacts 模块确实存在于这个版本上，它确实包含我正在搜索的方法。 Link to the Github module's __init__。

我正在使用 10.4 Databricks ML Runtime 开发 Databricks，它预装了上述版本的 mlflow。

【问题讨论】：

【解决方案1】：

mlflow.artifacts 是在 MLflow 1.25 中引入的。它实际上不在 1.24 中：https://github.com/mlflow/mlflow/blob/v1.24.0/mlflow/artifacts/__init__.py

我怀疑您使用 MLflow 1.25 或更高版本成功登录，并且您正在使用 MLflow 1.24 或更低版本阅读。

【讨论】：