如何使用对象 "mlflow.tracking.client.MlflowClient" 加载模型？

Question

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

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

Answer 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(...)

如何使用对象 "mlflow.tracking.client.MlflowClient" 加载模型？

How to load a model using the object "mlflow.tracking.client.MlflowClient"?

mlflow