时间戳

首页 > 解决方案 > MLflow：INVALID_PARAMETER_VALUE：模型注册表存储不支持 URI './mlruns'

python - MLflow：INVALID_PARAMETER_VALUE：模型注册表存储不支持 URI './mlruns'

问题描述

当我尝试在模型注册表中注册模型时出现此错误。有人可以帮助我吗？

RestException: INVALID_PARAMETER_VALUE: Unsupported URI './mlruns' for model registry store. 
Supported schemes are: ['postgresql', 'mysql', 'sqlite', 'mssql']. 
See https://www.mlflow.org/docs/latest/tracking.html#storage for how to setup a compatible server.

标签： pythonmlflow

解决方案

Mlflow 需要数据库作为模型注册表的数据存储所以你必须运行跟踪服务器，数据库作为后端存储并将模型记录到这个跟踪服务器。使用 DB 最简单的方法是使用 SQLite。

mlflow server \
    --backend-store-uri sqlite:///mlflow.db \
    --default-artifact-root ./artifacts \
    --host 0.0.0.0

并将 MLFLOW_TRACKING_URI 环境变量设置为http://localhost:5000或

mlflow.set_tracking_uri("http://localhost:5000")

到达 http://localhost:5000 后，您可以从 UI 或代码中注册已记录的模型。

推荐阅读