如何从 MLflow 模型注册表加载最新的模型版本？

How can I load the latest model version from MLflow model registry?

我可以使用 mlflow 客户端加载特定版本的模型：

import mlflow

model_version = 1

model = mlflow.pyfunc.load_model(
    model_uri=f"models:/c3760a15e6ac48f88ad7e5af940047d4/{model_version}"
)

但是有没有办法加载最新的模型版本？

没有像加载latest这样的东西，但是：

  • 您可以指定阶段 (staging、production) - 请参阅 docs
  • 您可以使用 get_latest_versions 函数找到最新版本 - 但它也会 return 每个阶段的最新版本

因此您需要定义 latest 对您意味着什么。