如何使用对象“mlflow.tracking.client.MlflowClient”加载模型？答案

【问题标题】：How to load a model using the object "mlflow.tracking.client.MlflowClient"?如何使用对象“mlflow.tracking.client.MlflowClient”加载模型？
【发布时间】：2021-01-20 18:46:15
【问题描述】：

我被 MLFlow 模型注册表卡住了。有谁知道如何使用“mlflow.tracking.client.MlflowClient”对象加载模型？

我想在此之后做一个预测。我确定我在某个地方错了，因为我过去已经这样做了。我无法在文档和网络中找到它。

【问题讨论】：

标签： mlflow

【解决方案1】：

您必须使用mlflow.<model_flavor>.load_model() 从模型注册表加载给定模型。例如：

import mlflow.pyfunc

model = mlflow.pyfunc.load_model(
          model_uri="models:/<model_name>/<model_version>"
          )

model.predict(...)

使用mlflow.tracking.client.MlflowClient，您可以从模型注册表中检索有关模型的元数据，但要检索实际模型，您需要使用mlflow.<model_flavor>.load_model。例如，您可以使用 MlflowClient 获取给定模型的下载 URI，然后使用 mlflow.<flavor>.load_model 检索该模型。

model_uri = client.get_model_version_download_uri("<model_name>", <version>)
model = mlflow.pyfunc.load_model(model_uri)

model.predict(...)

【讨论】：

谢谢。我意识到这不是我真正的问题。但你至少回答了我问的问题。谢谢你。