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

Question

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

import mlflow

model_version = 1

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

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

Answer 1

不存在像 load 这样的东西latest，但是：

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

所以你需要定义什么latest对你来说意味着什么。