MLflow：INVALID_PARAMETER_VALUE：模型注册表存储不支持的 URI

Question

当我尝试在模型注册表中注册模型时出现此错误。有人可以帮我吗？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.

Answer 1

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 或代码注册一个记录的模型。