如何获取MLflow中Production阶段模型版本?加载遇405错误
获取MLflow中Production阶段对应模型版本的方法
以下几种方法可以快速查到指定模型Production阶段对应的版本号:
MLflow CLI 命令
直接在终端执行以下命令,会返回目标模型Production阶段的版本详情:mlflow models get-latest-versions -n model_name -st ProductionPython API 调用
用MLflow的客户端库编写代码查询,适配1.23.1版本:from mlflow.tracking import MlflowClient # 初始化客户端(连接远程服务器需预先配置MLFLOW_TRACKING_URI) client = MlflowClient() # 获取Production阶段的模型版本列表 prod_model_versions = client.get_latest_versions("model_name", stages=["Production"]) # 遍历输出版本信息 for version in prod_model_versions: print(f"Production版本号: {version.version}")MLflow UI 查看
打开MLflow的Web管理界面,找到目标模型后切换到「Versions」标签页,即可看到标记为Production的模型版本号。
拿到版本号后,你就可以通过models:/model_name/版本号的方式正常加载模型,绕过当前遇到的405方法不允许错误。
内容的提问来源于stack exchange,提问作者Nourless
相关产品推荐
相关产品推荐

