REST API

MLflow REST API 允许您创建、列出和获取实验和运行,以及记录参数、指标和工件。 API 托管在 MLflow 跟踪服务器的 /api 路由下。例如,要在托管在 https://:5000 上的跟踪服务器上搜索实验,请向 https://:5000/api/2.0/mlflow/experiments/search 发送 POST 请求。


创建实验

端点

HTTP 方法

2.0/mlflow/experiments/create

POST

使用名称创建实验。返回新创建实验的 ID。验证具有相同名称的另一个实验是否已存在,如果具有相同名称的另一个实验已存在,则失败。

如果具有给定名称的实验已存在,则引发 RESOURCE_ALREADY_EXISTS

请求结构

字段名称

类型

描述

name

STRING

实验名称。此字段是必需的。

artifact_location

STRING

存储实验所有工件的位置。如果未提供,远程服务器将选择适当的默认值。

tags

一个 ExperimentTag 数组

要设置在实验上的一组标签。每个请求的标签最大大小和标签数量取决于存储后端。保证所有存储后端都支持最大大小为 250 字节的标签键和最大大小为 5000 字节的标签值。所有存储后端也保证每个请求最多支持 20 个标签。

响应结构

字段名称

类型

描述

experiment_id

STRING

实验的唯一标识符。


搜索实验

端点

HTTP 方法

2.0/mlflow/experiments/search

POST

请求结构

字段名称

类型

描述

max_results

INT64

所需的实验最大数量。服务器可能会选择所需的默认 max_results 值。所有服务器都保证支持至少为 1,000 的 max_results 阈值,但可能会支持更多。鼓励此端点的调用者明确传递 max_results 并利用 page_token 迭代实验。

page_token

STRING

指示要获取的实验页面的令牌

filter

STRING

一个关于实验属性和标签的过滤表达式,允许返回实验的子集。该语法是 SQL 的一个子集,支持在属性或标签与常量之间进行 AND 组合的二进制操作。

示例: name LIKE 'test-%' AND tags.key = 'value'

您可以使用双引号或反引号选择带有特殊字符(连字符、空格、句点等)的列。

示例: tags."extra-key" = 'value'tags.`extra-key` = 'value'

支持的运算符是 =!=LIKEILIKE

order_by

一个 STRING 数组

用于排序搜索结果的列列表,可以包括带有可选的“DESC”或“ASC”注释的实验名称和 ID,其中“ASC”是默认值。平局以实验 ID DESC 打破。

view_type

ViewType

要返回的实验类型的限定符。如果未指定,则仅返回活动实验。

响应结构

字段名称

类型

描述

experiments

一个 Experiment 数组

匹配搜索条件的实验

next_page_token

STRING

可用于检索下一页实验的令牌。空令牌表示没有更多实验可供检索。


获取实验

端点

HTTP 方法

2.0/mlflow/experiments/get

GET

获取实验的元数据。此方法适用于已删除的实验。

请求结构

字段名称

类型

描述

experiment_id

STRING

关联实验的 ID。此字段是必需的。

响应结构

字段名称

类型

描述

experiment

Experiment

实验详情。


按名称获取实验

端点

HTTP 方法

2.0/mlflow/experiments/get-by-name

GET

获取实验的元数据。

此端点将返回已删除的实验,但如果活动实验和已删除实验共享相同的名称,则优先返回活动实验。如果多个已删除的实验共享相同的名称,API 将返回其中一个。

如果不存在具有指定名称的实验,则引发 RESOURCE_DOES_NOT_EXIST

请求结构

字段名称

类型

描述

experiment_name

STRING

关联实验的名称。此字段是必需的。

响应结构

字段名称

类型

描述

experiment

Experiment

实验详情。


删除实验

端点

HTTP 方法

2.0/mlflow/experiments/delete

POST

标记实验和相关的元数据、运行、指标、参数和标签以供删除。如果实验使用 FileStore,则也会删除与实验关联的工件。

请求结构

字段名称

类型

描述

experiment_id

STRING

关联实验的 ID。此字段是必需的。


恢复实验

端点

HTTP 方法

2.0/mlflow/experiments/restore

POST

恢复标记为删除的实验。这也恢复了相关的元数据、运行、指标、参数和标签。如果实验使用 FileStore,也会恢复与实验关联的基础工件。

如果实验从未创建或已被永久删除,则引发 RESOURCE_DOES_NOT_EXIST

请求结构

字段名称

类型

描述

experiment_id

STRING

关联实验的 ID。此字段是必需的。


更新实验

端点

HTTP 方法

2.0/mlflow/experiments/update

POST

更新实验元数据。

请求结构

字段名称

类型

描述

experiment_id

STRING

关联实验的 ID。此字段是必需的。

new_name

STRING

如果提供,则实验的名称更改为新名称。新名称必须是唯一的。


创建运行

端点

HTTP 方法

2.0/mlflow/runs/create

POST

在实验中创建新运行。运行通常是机器学习或数据 ETL 流水线的一次单独执行。MLflow 使用运行来跟踪与单次执行相关的 ParamMetricRunTag

请求结构

字段名称

类型

描述

experiment_id

STRING

关联实验的 ID。

user_id

STRING

执行运行的用户的 ID。此字段自 MLflow 1.0 起已弃用,未来版本的 MLflow 中将删除。请使用“mlflow.user”标签。

run_name

STRING

运行的名称。

start_time

INT64

运行开始时间的毫秒级 Unix 时间戳。

tags

一个 RunTag 数组

运行的其他元数据。

响应结构

字段名称

类型

描述

run

Run

新创建的运行。


删除运行

端点

HTTP 方法

2.0/mlflow/runs/delete

POST

标记运行以供删除。

请求结构

字段名称

类型

描述

run_id

STRING

要删除的运行的 ID。此字段是必需的。


恢复运行

端点

HTTP 方法

2.0/mlflow/runs/restore

POST

恢复已删除的运行。这也恢复了关联的元数据、运行、指标、参数和标签。如果实验使用 FileStore,也会恢复与实验关联的基础工件。

请求结构

字段名称

类型

描述

run_id

STRING

要恢复的运行的 ID。此字段是必需的。


获取运行

端点

HTTP 方法

2.0/mlflow/runs/get

GET

获取运行的元数据、指标、参数和标签。在为运行记录了具有相同键的多个指标的情况下,仅返回具有最新时间戳的值。如果多个值具有最新时间戳,则返回这些值中的最大值。

请求结构

字段名称

类型

描述

run_id

STRING

要获取的运行的 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 要获取的运行的 ID。此字段将在未来的 MLflow 版本中删除。

响应结构

字段名称

类型

描述

run

Run

运行元数据(名称、开始时间等)和数据(指标、参数和标签)。


记录指标

端点

HTTP 方法

2.0/mlflow/runs/log-metric

POST

为运行记录一个指标。指标是具有关联时间戳的键值对(字符串键,浮点值)。示例包括表示 ML 模型准确性的各种指标。可以为运行多次记录指标。

请求结构

字段名称

类型

描述

run_id

STRING

记录指标的运行 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 记录指标的运行 ID。此字段将在未来的 MLflow 版本中删除。

key

STRING

指标的名称。此字段是必需的。

value

DOUBLE

记录的指标的浮点值。此字段是必需的。

timestamp

INT64

记录指标时的毫秒级 Unix 时间戳。此字段是必需的。

step

INT64

记录指标的步长


批量记录

端点

HTTP 方法

2.0/mlflow/runs/log-batch

POST

为运行记录一批指标、参数和标签。如果任何数据未能持久化,服务器将以错误(非 200 状态码)响应。在发生错误(由于内部服务器错误或无效请求)的情况下,可能会写入部分数据。

您可以交错记录指标、参数和标签,但在给定的实体类型内,保证遵循请求主体中指定的顺序。也就是说,对于像这样的 API 请求

{
   "run_id": "2a14ed5c6a87499199e0106c3501eab8",
   "metrics": [
     {"key": "mae", "value": 2.5, "timestamp": 1552550804},
     {"key": "rmse", "value": 2.7, "timestamp": 1552550804},
   ],
   "params": [
     {"key": "model_class", "value": "LogisticRegression"},
   ]
}

服务器保证在“mae”之后写入指标“rmse”,但它可能在两个指标之前、之后或之后写入参数“model_class”。

指标、参数和标签的覆盖行为如下

  • 指标:指标值永远不会被覆盖。记录一个指标(键、值、时间戳)会追加到具有提供键的指标的值集。

  • 标签:标签值可以被对同一标签键的连续写入覆盖。也就是说,如果在同一 API 请求中提供了多个具有相同键的标签值,则写入最后提供的标签值。记录相同的标签(键、值)是允许的——也就是说,记录标签是幂等的。

  • 参数:一旦写入,参数值就不能更改(尝试覆盖参数值将导致错误)。但是,记录相同的参数(键、值)是允许的——也就是说,记录参数是幂等的。

请求限制

单个 JSON 序列化的 API 请求大小可能高达 1 MB,并包含

  • 总共不超过 1000 个指标、参数和标签

  • 最多 1000 个指标

  • 最多 100 个参数

  • 最多 100 个标签

例如,一个有效的请求可能包含 900 个指标、50 个参数和 50 个标签,但记录 900 个指标、50 个参数和 51 个标签是无效的。以下限制也适用于指标、参数和标签的键和值

  • 指标、参数和标签的键最长可达 250 个字符

  • 参数和标签的值最长可达 250 个字符

请求结构

字段名称

类型

描述

run_id

STRING

要记录的运行 ID

metrics

一个 Metric 数组

要记录的指标。单个请求最多可包含 1000 个指标,以及总共最多 1000 个指标、参数和标签。

params

一个 Param 数组

要记录的参数。单个请求最多可包含 100 个参数,以及总共最多 1000 个指标、参数和标签。

tags

一个 RunTag 数组

要记录的标签。单个请求最多可包含 100 个标签,以及总共最多 1000 个指标、参数和标签。


记录模型

端点

HTTP 方法

2.0/mlflow/runs/log-model

POST

注意

实验性:此 API 可能会在未来版本中更改或删除,恕不另行通知。

请求结构

字段名称

类型

描述

run_id

STRING

要记录的运行 ID

model_json

STRING

MLmodel 文件,JSON 格式。


记录输入

端点

HTTP 方法

2.0/mlflow/runs/log-inputs

POST

请求结构

字段名称

类型

描述

run_id

STRING

要记录的运行 ID 此字段是必需的。

datasets

一个 DatasetInput 数组

数据集输入


设置实验标签

端点

HTTP 方法

2.0/mlflow/experiments/set-experiment-tag

POST

为实验设置标签。实验标签是可更新的元数据。

请求结构

字段名称

类型

描述

experiment_id

STRING

记录标签的实验 ID。必须提供。此字段是必需的。

key

STRING

标签的名称。最大大小取决于存储后端。所有存储后端都保证支持最大大小为 250 字节的键值。此字段是必需的。

value

STRING

记录的标签的字符串值。最大大小取决于存储后端。所有存储后端都保证支持最大大小为 5000 字节的键值。此字段是必需的。


删除实验标签

端点

HTTP 方法

2.0/mlflow/experiments/delete-experiment-tag

POST

删除实验上的标签。

请求结构

字段名称

类型

描述

experiment_id

STRING

记录标签的实验 ID。必须提供。此字段是必需的。

key

STRING

标签的名称。最大大小为 255 字节。必须提供。此字段是必需的。


设置标签

端点

HTTP 方法

2.0/mlflow/runs/set-tag

POST

为运行设置标签。标签是可以在运行期间和运行完成后更新的运行元数据。

请求结构

字段名称

类型

描述

run_id

STRING

记录标签的运行 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 记录标签的运行 ID。此字段将在未来的 MLflow 版本中删除。

key

STRING

标签的名称。最大大小取决于存储后端。所有存储后端都保证支持最大大小为 250 字节的键值。此字段是必需的。

value

STRING

记录的标签的字符串值。最大大小取决于存储后端。所有存储后端都保证支持最大大小为 5000 字节的键值。此字段是必需的。


删除标签

端点

HTTP 方法

2.0/mlflow/runs/delete-tag

POST

删除运行上的标签。标签是可以在运行期间和运行完成后更新的运行元数据。

请求结构

字段名称

类型

描述

run_id

STRING

记录标签的运行 ID。必须提供。此字段是必需的。

key

STRING

标签的名称。最大大小为 255 字节。必须提供。此字段是必需的。


记录参数

端点

HTTP 方法

2.0/mlflow/runs/log-parameter

POST

为运行记录一个参数。参数是键值对(字符串键,字符串值)。示例包括用于 ML 模型训练的超参数以及 ETL 流水线中使用的常量日期和值。一个参数只能为运行记录一次。

请求结构

字段名称

类型

描述

run_id

STRING

记录参数的运行 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 记录参数的运行 ID。此字段将在未来的 MLflow 版本中删除。

key

STRING

参数的名称。最大大小为 255 字节。此字段是必需的。

value

STRING

记录的参数的字符串值。最大大小为 6000 字节。此字段是必需的。


获取指标历史记录

端点

HTTP 方法

2.0/mlflow/metrics/get-history

GET

获取给定运行中指定指标的所有值的列表。

请求结构

字段名称

类型

描述

run_id

STRING

要从中获取指标值的运行的 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 要从中获取指标值的运行的 ID。此字段将在未来的 MLflow 版本中删除。

metric_key

STRING

指标的名称。此字段是必需的。

page_token

STRING

指示要获取的指标历史记录页面的令牌

max_results

INT32

每调返回的运行中一个指标的记录实例的最大数量。后端服务器可能会根据性能要求限制 max_results 的值。未指定此值的请求将表现为非分页查询,其中返回单个响应中的给定运行中给定指标的所有指标历史值。

响应结构

字段名称

类型

描述

metrics

一个 Metric 数组

此指标的所有记录值。

next_page_token

STRING

可用于查询下一页指标历史记录值的令牌。缺失的令牌表示没有额外的指标可供获取。


搜索运行

端点

HTTP 方法

2.0/mlflow/runs/search

POST

搜索满足表达式的运行。搜索表达式可以使用 MetricParam 键。

请求结构

字段名称

类型

描述

experiment_ids

一个 STRING 数组

要在其上搜索的实验 ID 列表。

filter

STRING

一个关于参数、指标和标签的过滤表达式,允许返回运行的子集。该语法是 SQL 的一个子集,支持在参数、指标或标签与常量之间进行 AND 组合的二进制操作。

示例: metrics.rmse < 1 and params.model_class = 'LogisticRegression'

您可以使用双引号选择带有特殊字符(连字符、空格、句点等)的列: metrics."model class" = 'LinearRegression' and tags."user-name" = 'Tomas'

支持的运算符是 =!=>>=<<=

run_view_type

ViewType

是仅显示活动运行、仅显示已删除运行还是显示所有运行。默认为仅显示活动运行。

max_results

INT32

所需的运行最大数量。如果未指定,默认为 1000。所有服务器都保证支持至少为 50,000 的 max_results 阈值,但可能会支持更多。鼓励此端点的调用者明确传递 max_results 并利用 page_token 迭代实验。

order_by

一个 STRING 数组

要排序的列列表,包括带有可选的“DESC”或“ASC”注释的属性、参数、指标和标签,其中“ASC”是默认值。示例:[“params.input DESC”, “metrics.alpha ASC”, “metrics.rmse”] 平局以 start_time DESC 打破,然后是具有相同开始时间的运行的 run_id(如果未提供 order_by,这是默认排序标准)。

page_token

STRING

响应结构

字段名称

类型

描述

runs

一个 Run 数组

匹配搜索条件的运行。

next_page_token

STRING


列出工件

端点

HTTP 方法

2.0/mlflow/artifacts/list

GET

列出运行的工件。接受可选的 artifact_path 前缀,如果指定,则响应仅包含具有指定前缀的工件。

请求结构

字段名称

类型

描述

run_id

STRING

要列出其工件的运行 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 要列出其工件的运行 ID。此字段将在未来的 MLflow 版本中删除。

path

STRING

过滤与此路径匹配的工件(从根工件目录的相对路径)。

page_token

STRING

指示要获取的工件结果页面的令牌

响应结构

字段名称

类型

描述

root_uri

STRING

运行的根工件目录。

files

一个 FileInfo 数组

工件的文件位置和元数据。

next_page_token

STRING

可用于检索下一页工件结果的令牌


更新运行

端点

HTTP 方法

2.0/mlflow/runs/update

POST

更新运行元数据。

请求结构

字段名称

类型

描述

run_id

STRING

要更新的运行 ID。必须提供。

run_uuid

STRING

[已弃用,请改用 run_id] 要更新的运行的 ID.。此字段将在未来的 MLflow 版本中删除。

status

RunStatus

运行的更新后的状态。

end_time

INT64

运行结束时间的毫秒级 Unix 时间戳。

run_name

STRING

运行的新名称。

响应结构

字段名称

类型

描述

run_info

RunInfo

更新运行的元数据。


列出评分器

端点

HTTP 方法

2.0/mlflow/scorers/list

GET

列出实验的所有评分器。

请求结构

列出实验的所有评分器。

字段名称

类型

描述

experiment_id

STRING

实验 ID。

响应结构

字段名称

类型

描述

scorers

一个 Scorer 实体的数组

评分器实体列表(每个评分器名称的最新版本)。


列出评分器版本

端点

HTTP 方法

2.0/mlflow/scorers/versions

GET

列出实验中特定评分器的所有版本。

请求结构

列出实验中特定评分器的所有版本。

字段名称

类型

描述

experiment_id

STRING

实验 ID。

name

STRING

评分器名称。

响应结构

字段名称

类型

描述

scorers

一个 Scorer 实体的数组

评分器的所有版本的评分器实体列表。


注册评分器

端点

HTTP 方法

2.0/mlflow/scorers/register

POST

为实验注册一个评分器。

请求结构

为实验注册一个评分器。

字段名称

类型

描述

experiment_id

STRING

实验 ID。

name

STRING

评分器名称。

serialized_scorer

STRING

序列化的评分器字符串(JSON)。

响应结构

字段名称

类型

描述

version

INT32

评分器的新版本号。

scorer_id

STRING

评分器的唯一标识符。

experiment_id

STRING

实验 ID(与请求相同)。

name

STRING

评分器名称(与请求相同)。

serialized_scorer

STRING

序列化的评分器字符串(与请求相同)。

creation_time

INT64

评分器版本的创建时间(自纪元以来的毫秒数)。


获取评分器

端点

HTTP 方法

2.0/mlflow/scorers/get

GET

获取实验的特定评分器。

请求结构

获取实验的特定评分器。

字段名称

类型

描述

experiment_id

STRING

实验 ID。

name

STRING

评分器名称。

version

INT32

评分器版本。如果未指定,则返回具有最大版本的评分器。

响应结构

字段名称

类型

描述

scorer

评分器

评分器实体。


删除评分器

端点

HTTP 方法

2.0/mlflow/scorers/delete

DELETE

删除实验的评分器。

请求结构

删除实验的评分器。

字段名称

类型

描述

experiment_id

STRING

实验 ID。

name

STRING

评分器名称。

version

INT32

要删除的评分器版本。如果未指定,则删除所有版本。


创建网关端点

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/create

POST

使用模型配置创建新端点

请求结构

字段名称

类型

描述

name

STRING

端点的可选用户友好名称

model_configs

一个 GatewayEndpointModelConfig 实体的数组

模型配置列表

created_by

STRING

创建者的用户名

routing_strategy

RoutingStrategy

端点的可选路由策略

fallback_config

FallbackConfig

可选的备用配置(包括策略、最大重试次数)

响应结构

字段名称

类型

描述

endpoint

GatewayEndpoint

带有所有模型映射的已创建端点


更新网关端点

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/update

POST

更新端点的名称

请求结构

字段名称

类型

描述

endpoint_id

STRING

要更新的端点的 ID

name

STRING

端点的可选新名称

updated_by

STRING

更新者的用户名

model_configs

一个 GatewayEndpointModelConfig 实体的数组

可选的新模型配置列表(替换所有现有的模型链接)

routing_strategy

RoutingStrategy

端点的可选新路由策略

fallback_config

FallbackConfig

可选的备用配置(包括策略、最大重试次数)

响应结构

字段名称

类型

描述

endpoint

GatewayEndpoint

已更新的端点


删除网关端点

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/delete

DELETE

删除端点及其所有模型配置

请求结构

字段名称

类型

描述

endpoint_id

STRING

要删除的端点的 ID


获取网关端点

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/get

GET

获取端点详细信息,包括所有模型配置

请求结构

字段名称

类型

描述

endpoint_id

STRING

必须提供 endpoint_id 或 name

name

STRING

响应结构

字段名称

类型

描述

endpoint

GatewayEndpoint

带有所有模型配置的端点


列出网关端点

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/list

GET

列出端点,并可按提供者或密钥进行可选过滤

请求结构

字段名称

类型

描述

provider

STRING

按提供者进行可选过滤

secret_id

STRING

按密钥 ID 进行可选过滤

响应结构

字段名称

类型

描述

endpoints

一个 GatewayEndpoint 实体的数组

带有其模型配置的端点列表


创建端点绑定

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/bindings/create

POST

在端点和 MLflow 资源之间创建绑定

请求结构

字段名称

类型

描述

endpoint_id

STRING

要绑定的端点的 ID

resource_type

STRING

MLflow 资源的类型

resource_id

STRING

资源实例的 ID

created_by

STRING

创建者的用户名

响应结构

字段名称

类型

描述

binding

GatewayEndpointBinding

已创建的绑定


删除端点绑定

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/bindings/delete

DELETE

删除端点和资源之间的绑定

请求结构

字段名称

类型

描述

endpoint_id

STRING

端点的 ID

resource_type

STRING

绑定到端点的资源类型

resource_id

STRING

资源的 ID


列出端点绑定

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/bindings/list

GET

列出端点的所有绑定

请求结构

字段名称

类型

描述

endpoint_id

STRING

要列出绑定的端点的 ID

resource_type

STRING

用于过滤绑定的资源类型(例如,“scorer”)

resource_id

STRING

用于过滤绑定的资源 ID

响应结构

字段名称

类型

描述

bindings

一个 GatewayEndpointBinding 实体的数组

端点的绑定列表


创建网关模型定义

端点

HTTP 方法

2.0/mlflow/gateway/model-definitions/create

POST

创建可重用的模型定义

请求结构

字段名称

类型

描述

name

STRING

模型定义的友好名称(必须唯一)

secret_id

STRING

包含身份验证凭据的密钥的 ID

provider

STRING

LLM 提供商(例如,“openai”、“anthropic”)

model_name

STRING

提供商特定的模型标识符(例如,“gpt-4o”、“claude-3-5-sonnet”)

created_by

STRING

创建者的用户名

响应结构

字段名称

类型

描述

model_definition

GatewayModelDefinition

已创建的模型定义


更新网关模型定义

端点

HTTP 方法

2.0/mlflow/gateway/model-definitions/update

POST

更新模型定义

请求结构

字段名称

类型

描述

model_definition_id

STRING

要更新的模型定义的 ID

name

STRING

可选的新名称

secret_id

STRING

可选的新密钥 ID

model_name

STRING

可选的新模型名称

updated_by

STRING

更新者的用户名

provider

STRING

可选的新提供商

响应结构

字段名称

类型

描述

model_definition

GatewayModelDefinition

已更新的模型定义


删除网关模型定义

端点

HTTP 方法

2.0/mlflow/gateway/model-definitions/delete

DELETE

删除模型定义(如果被任何端点使用则失败)

请求结构

字段名称

类型

描述

model_definition_id

STRING

要删除的模型定义的 ID(如果被任何端点使用则失败)


获取网关模型定义

端点

HTTP 方法

2.0/mlflow/gateway/model-definitions/get

GET

按 ID 获取模型定义

请求结构

字段名称

类型

描述

model_definition_id

STRING

要检索的模型定义的 ID

响应结构

字段名称

类型

描述

model_definition

GatewayModelDefinition

模型定义


列出网关模型定义

端点

HTTP 方法

2.0/mlflow/gateway/model-definitions/list

GET

列出所有模型定义,并可进行可选过滤

请求结构

字段名称

类型

描述

provider

STRING

按提供者进行可选过滤

secret_id

STRING

按密钥 ID 进行可选过滤

响应结构

字段名称

类型

描述

model_definitions

一个 GatewayModelDefinition 实体的数组

模型定义列表


创建网关密钥

端点

HTTP 方法

2.0/mlflow/gateway/secrets/create

POST

创建用于 LLM 提供商身份验证的新加密密钥

请求结构

字段名称

类型

描述

secret_name

STRING

密钥的友好名称(必须唯一)

secret_value

一个 SecretValueEntry 实体的数组

要加密的密钥值,作为键值对。对于简单的 API 密钥:{“api_key”: “sk-xxx”} 对于复合凭据:{“aws_access_key_id”: “...”, “aws_secret_access_key”: “...”}

provider

STRING

可选的 LLM 提供商(例如,“openai”、“anthropic”)

auth_config

一个 AuthConfigEntry 实体的数组

可选的提供商特定身份验证配置。对于多身份验证提供商,请包含“auth_mode”键(例如,{“auth_mode”: “access_keys”, “aws_region_name”: “us-east-1”})

created_by

STRING

创建者的用户名

响应结构

字段名称

类型

描述

secret

GatewaySecretInfo

已创建的密钥元数据(不包含加密值)


更新网关密钥

端点

HTTP 方法

2.0/mlflow/gateway/secrets/update

POST

更新现有密钥的值或身份验证配置

请求结构

字段名称

类型

描述

secret_id

STRING

要更新的密钥的 ID

secret_value

一个 SecretValueEntry 实体的数组

用于密钥轮换的可选新密钥值,作为键值对(空映射 = 无更改)。对于简单的 API 密钥:{“api_key”: “sk-xxx”} 对于复合凭据:{“aws_access_key_id”: “...”, “aws_secret_access_key”: “...”}

auth_config

一个 AuthConfigEntry 实体的数组

可选的新身份验证配置。对于多身份验证提供商,请包含“auth_mode”键(例如,{“auth_mode”: “access_keys”, “aws_region_name”: “us-east-1”})

updated_by

STRING

更新者的用户名

响应结构

字段名称

类型

描述

secret

GatewaySecretInfo

已更新的密钥元数据


删除网关密钥

端点

HTTP 方法

2.0/mlflow/gateway/secrets/delete

DELETE

删除密钥

请求结构

字段名称

类型

描述

secret_id

STRING

要删除的密钥的 ID


获取网关密钥信息

端点

HTTP 方法

2.0/mlflow/gateway/secrets/get

GET

获取关于密钥的元数据(不包含加密值)

请求结构

字段名称

类型

描述

secret_id

STRING

必须提供 secret_id 或 secret_name

secret_name

STRING

响应结构

字段名称

类型

描述

secret

GatewaySecretInfo

密钥元数据(不包含加密值)


列出网关密钥

端点

HTTP 方法

2.0/mlflow/gateway/secrets/list

GET

列出所有密钥,并可按提供者进行可选过滤

请求结构

字段名称

类型

描述

provider

STRING

按提供者进行可选过滤(例如,“openai”、“anthropic”)

响应结构

字段名称

类型

描述

secrets

一个 GatewaySecretInfo 实体的数组

密钥元数据列表(不包含加密值)


将模型附加到端点

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/models/attach

POST

将现有的模型定义附加到端点

请求结构

字段名称

类型

描述

endpoint_id

STRING

要将模型附加到的端点的 ID

model_config

GatewayEndpointModelConfig

要附加模型的配置

created_by

STRING

创建者的用户名

响应结构

字段名称

类型

描述

mapping

GatewayEndpointModelMapping

已创建的映射


从端点分离模型

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/models/detach

POST

将模型定义从端点分离(不删除模型定义)

请求结构

字段名称

类型

描述

endpoint_id

STRING

端点的 ID

model_definition_id

STRING

要分离的模型定义的 ID


网关设置端点标签

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/set-tag

POST

在端点上设置标签

请求结构

字段名称

类型

描述

endpoint_id

STRING

要设置标签的端点的 ID

key

STRING

要设置的标签键

value

STRING

要设置的标签值


网关删除端点标签

端点

HTTP 方法

2.0/mlflow/gateway/endpoints/delete-tag

DELETE

从端点删除标签

请求结构

字段名称

类型

描述

endpoint_id

STRING

要删除标签的端点的 ID

key

STRING

要删除的标签键


创建已注册模型

端点

HTTP 方法

2.0/mlflow/registered-models/create

POST

如果具有给定名称的已注册模型已存在,则抛出 RESOURCE_ALREADY_EXISTS

请求结构

字段名称

类型

描述

name

STRING

在此名称下注册模型。此字段是必需的。

tags

一个 RegisteredModelTag 实体的数组

已注册模型的附加元数据。

description

STRING

已注册模型的可选描述。

deployment_job_id

STRING

此模型的部署作业 ID。

响应结构

字段名称

类型

描述

registered_model

RegisteredModel


获取已注册模型

端点

HTTP 方法

2.0/mlflow/registered-models/get

GET

请求结构

字段名称

类型

描述

name

STRING

已注册模型的唯一名称标识符。此字段是必需的。

响应结构

字段名称

类型

描述

registered_model

RegisteredModel


重命名已注册模型

端点

HTTP 方法

2.0/mlflow/registered-models/rename

POST

请求结构

字段名称

类型

描述

name

STRING

已注册模型的唯一名称标识符。此字段是必需的。

new_name

STRING

如果提供,则更新此 registered_model 的名称。

响应结构

字段名称

类型

描述

registered_model

RegisteredModel


更新已注册模型

端点

HTTP 方法

2.0/mlflow/registered-models/update

PATCH

请求结构

字段名称

类型

描述

name

STRING

已注册模型的唯一名称标识符。此字段是必需的。

description

STRING

如果提供,则更新此 registered_model 的描述。

deployment_job_id

STRING

此模型的部署作业 ID。

响应结构

字段名称

类型

描述

registered_model

RegisteredModel


删除已注册模型

端点

HTTP 方法

2.0/mlflow/registered-models/delete

DELETE

请求结构

字段名称

类型

描述

name

STRING

已注册模型的唯一名称标识符。此字段是必需的。


获取最新的模型版本

端点

HTTP 方法

2.0/mlflow/registered-models/get-latest-versions

POST

请求结构

字段名称

类型

描述

name

STRING

已注册模型的唯一名称标识符。此字段是必需的。

stages

一个 STRING 数组

阶段列表。

响应结构

字段名称

类型

描述

model_versions

一个 ModelVersion 实体的数组

每个请求阶段的最新版本模型。仅返回状态为 READY 的模型。如果未提供 stages,则返回每个阶段的最新版本,包括 "None"


创建模型版本

端点

HTTP 方法

2.0/mlflow/model-versions/create

POST

请求结构

字段名称

类型

描述

name

STRING

在此名称下注册模型。此字段是必需的。

source

STRING

指示模型工件位置的 URI。此字段是必需的。

run_id

STRING

MLflow 运行 ID 用于关联,如果 source 是由 MLflow 跟踪服务器中的实验运行生成的

tags

一个 ModelVersionTag 实体的数组

模型版本的附加元数据。

run_link

STRING

MLflow 运行链接 - 这是生成此模型版本的运行的确切链接,可能托管在另一个 MLflow 实例上。

description

STRING

模型版本可选描述。

model_id

STRING

模型版本的可选 model_id,用于将已注册模型链接到源已记录的模型

响应结构

字段名称

类型

描述

model_version

ModelVersion

返回注册表中为此模型生成的新版本号。


获取模型版本

端点

HTTP 方法

2.0/mlflow/model-versions/get

GET

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。

响应结构

字段名称

类型

描述

model_version

ModelVersion


更新模型版本

端点

HTTP 方法

2.0/mlflow/model-versions/update

PATCH

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。

description

STRING

如果提供,则更新此 registered_model 的描述。

响应结构

字段名称

类型

描述

model_version

ModelVersion

返回注册表中为此模型生成的新版本号。


删除模型版本

端点

HTTP 方法

2.0/mlflow/model-versions/delete

DELETE

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。


搜索模型版本

端点

HTTP 方法

2.0/mlflow/model-versions/search

GET

请求结构

字段名称

类型

描述

filter

STRING

字符串过滤条件,例如“name=’my-model-name’” 。必须是单个布尔条件,字符串值用单引号括起来。

max_results

INT64

所需模型的最大数量。最大阈值为 200K。后端可以选择较低的默认值和最大阈值。

order_by

一个 STRING 数组

要排序的列列表,包括模型名称、版本、阶段,并带有可选的“DESC”或“ASC”注释,其中“ASC”是默认值。平局按最新阶段转换时间戳、然后按名称 ASC、然后按版本 DESC 进行。

page_token

STRING

用于根据之前的搜索查询转到下一页的分页令牌。

响应结构

字段名称

类型

描述

model_versions

一个 ModelVersion 实体的数组

符合搜索条件的模型

next_page_token

STRING

用于为同一搜索查询请求下一页模型的令牌。


获取模型版本工件的下载 URI

端点

HTTP 方法

2.0/mlflow/model-versions/get-download-uri

GET

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。

响应结构

字段名称

类型

描述

artifact_uri

STRING

存储此模型版本工件的 URI。


转换模型版本阶段

端点

HTTP 方法

2.0/mlflow/model-versions/transition-stage

POST

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。

stage

STRING

model_version 转换到新阶段。此字段是必需的。

archive_existing_versions

布尔值

在将模型版本转换为特定阶段时,此标志决定是否应将该阶段所有现有的模型版本原子地移至“已存档”阶段。这确保了目标阶段最多只有一个模型版本。此字段在转换模型版本的阶段时是*必需的*。此字段是必需的。

响应结构

字段名称

类型

描述

model_version

ModelVersion

已更新的模型版本


搜索已注册模型

端点

HTTP 方法

2.0/mlflow/registered-models/search

GET

请求结构

字段名称

类型

描述

filter

STRING

字符串过滤条件,例如“name LIKE ‘my-model-name’” 。后端会自动解释为“name LIKE ‘%my-model-name%’” 。单个布尔条件,字符串值用单引号括起来。

max_results

INT64

所需模型的最大数量。默认为 100。最大阈值为 1000。

order_by

一个 STRING 数组

用于排序搜索结果的列列表,包括模型名称和最后更新时间戳,并带有可选的“DESC”或“ASC”注释,其中“ASC”是默认值。平局按模型名称 ASC 进行。

page_token

STRING

用于根据之前的搜索查询转到下一页的分页令牌。

响应结构

字段名称

类型

描述

registered_models

一个 RegisteredModel 实体的数组

符合搜索条件的已注册模型。

next_page_token

STRING

用于请求下一页模型的令牌。


设置已注册模型标签

端点

HTTP 方法

2.0/mlflow/registered-models/set-tag

POST

请求结构

字段名称

类型

描述

name

STRING

模型的唯一名称。此字段是必需的。

key

STRING

标签的名称。最大大小取决于存储后端。如果具有此名称的标签已存在,其先前的值将被指定的 value 替换。所有存储后端都保证支持最大 250 字节的键值。此字段是必需的。

value

STRING

正在记录的标签的字符串值。最大大小取决于存储后端。此字段是必需的。


设置模型版本标签

端点

HTTP 方法

2.0/mlflow/model-versions/set-tag

POST

请求结构

字段名称

类型

描述

name

STRING

模型的唯一名称。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。

key

STRING

标签的名称。最大大小取决于存储后端。如果具有此名称的标签已存在,其先前的值将被指定的 value 替换。所有存储后端都保证支持最大 250 字节的键值。此字段是必需的。

value

STRING

正在记录的标签的字符串值。最大大小取决于存储后端。此字段是必需的。


删除已注册模型标签

端点

HTTP 方法

2.0/mlflow/registered-models/delete-tag

DELETE

请求结构

字段名称

类型

描述

name

STRING

记录了标签的已注册模型的名称。此字段是必需的。

key

STRING

标签的名称。名称必须完全匹配;不支持通配符删除。最大大小为 250 字节。此字段是必需的。


删除模型版本标签

端点

HTTP 方法

2.0/mlflow/model-versions/delete-tag

DELETE

请求结构

字段名称

类型

描述

name

STRING

记录了标签的已注册模型的名称。此字段是必需的。

version

STRING

记录模型的版本号。此字段是必需的。

key

STRING

标签的名称。名称必须完全匹配;不支持通配符删除。最大大小为 250 字节。此字段是必需的。


删除已注册模型别名

端点

HTTP 方法

2.0/mlflow/registered-models/alias

DELETE

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

alias

STRING

别名的名称。名称必须精确匹配;不支持通配符删除。最大大小为 256 字节。此字段是必需的。


通过别名获取模型版本

端点

HTTP 方法

2.0/mlflow/registered-models/alias

GET

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

alias

STRING

别名的名称。最大大小为 256 字节。此字段是必需的。

响应结构

字段名称

类型

描述

model_version

ModelVersion


设置已注册模型别名

端点

HTTP 方法

2.0/mlflow/registered-models/alias

POST

请求结构

字段名称

类型

描述

name

STRING

已注册模型的名称。此字段是必需的。

alias

STRING

别名的名称。最大大小取决于存储后端。如果具有此名称的别名已存在,其预先存在的值将被指定的 version 替换。所有存储后端都保证支持大小高达 256 字节的别名名称值。此字段是必需的。

version

STRING

模型版本号。此字段是必需的。

数据结构

AuthConfigEntry

字段名称

类型

描述

key

STRING

value

STRING

AuthConfigEntry

字段名称

类型

描述

key

STRING

value

STRING

AuthConfigEntry

字段名称

类型

描述

key

STRING

value

STRING

Dataset

Dataset. 代表在模型开发过程中用于训练、测试或评估的数据引用。

字段名称

类型

描述

name

STRING

数据集的名称。例如“my.uc.table@2”、“nyc-taxi-dataset”、“fantastic-elk-3” 此字段是必需的。

digest

STRING

数据集摘要,例如数据集的 md5 散列,它在具有相同名称的数据集中唯一标识该数据集。此字段是必需的。

source_type

STRING

数据集来源的类型,例如“databricks-uc-table”、“DBFS”、“S3”等。此字段是必需的。

source

STRING

数据集的来源信息。请注意,如果数据集在使用 MLflow 之前经过了转换/修改,则来源可能无法完全重现该数据集。此字段是必需的。

schema

STRING

数据集的模式。例如,数据框的 MLflow ColSpec JSON、ndarray 的 MLflow TensorSpec JSON 或其他模式格式。

profile

STRING

数据集的概况。数据集的摘要统计信息,例如表中行的数量、表中每列的平均值/标准差/众数,或数组中元素的数量。

DatasetInput

DatasetInput。代表一个数据集和输入标签。

字段名称

类型

描述

tags

一个 InputTag 数组

数据集输入的标签列表,例如值为“training”的“context”标签

dataset

数据集

用作 Run 输入的数据集。此字段是必需的。

Experiment

Experiment

字段名称

类型

描述

experiment_id

STRING

实验的唯一标识符。

name

STRING

用于标识实验的人类可读名称。

artifact_location

STRING

存储实验工件的位置。

lifecycle_stage

STRING

实验当前的生命周期阶段:“active”(活动)或“deleted”(已删除)。已删除的实验不会由 API 返回。

last_update_time

INT64

最后更新时间

creation_time

INT64

创建时间

tags

一个 ExperimentTag 数组

标签:额外的元数据键值对。

ExperimentTag

实验的标签。

字段名称

类型

描述

key

STRING

标签键。

value

STRING

标签值。

FallbackConfig

回退路由的配置

字段名称

类型

描述

strategy

FallbackStrategy

回退策略。

max_attempts

INT32

回退路由的最大尝试次数(不能超过目标数量)。

FileInfo

字段名称

类型

描述

path

STRING

相对于运行的根工件目录的路径。

is_dir

布尔值

路径是否为目录。

file_size

INT64

大小(以字节为单位)。目录则未设置。

GatewayEndpoint

代表 LLM 网关端点的端点实体

字段名称

类型

描述

endpoint_id

STRING

端点的唯一标识符

name

STRING

端点的用户友好名称

created_at

INT64

端点创建时的时间戳(自纪元以来的毫秒数)

last_updated_at

INT64

端点最后更新时的时间戳(自纪元以来的毫秒数)

model_mappings

一个 GatewayEndpointModelMapping 数组

绑定到此端点的模型映射列表

created_by

STRING

创建端点的用户 ID

last_updated_by

STRING

最后更新端点的用户 ID

tags

一个 GatewayEndpointTag 数组

与端点关联的标签

routing_strategy

RoutingStrategy

端点的路由策略

fallback_config

FallbackConfig

回退配置(如果 routing_strategy 为 FALLBACK 则填充)

GatewayEndpointBinding

端点与 MLflow 资源之间的绑定。使用复合键(endpoint_id、resource_type、resource_id)进行标识。

字段名称

类型

描述

endpoint_id

STRING

此绑定引用的端点的 ID

resource_type

STRING

MLflow 资源的类型(例如,“scorer”)

resource_id

STRING

特定资源实例的 ID

created_at

INT64

创建绑定时的时间戳(自纪元以来的毫秒数)

last_updated_at

INT64

最后更新绑定时的时间戳(自纪元以来的毫秒数)

created_by

STRING

创建绑定的用户 ID

last_updated_by

STRING

最后更新绑定的用户 ID

display_name

STRING

字段 8-9 保留 - endpoint_name 和 model_mappings 已移除(客户端连接)资源的具有人类可读的显示名称(例如,评分器名称)

GatewayEndpointModelConfig

附加到端点的模型的配置

字段名称

类型

描述

model_definition_id

STRING

模型定义的 ID

linkage_type

GatewayModelLinkageType

链接类型

weight

FLOAT

用于流量分配的回退权重

fallback_order

INT32

回退尝试的顺序(仅适用于 FALLBACK 链接,PRIMARY 为 NULL)

GatewayEndpointModelMapping

端点与模型定义之间的映射

字段名称

类型

描述

mapping_id

STRING

此映射的唯一标识符

endpoint_id

STRING

端点的 ID

model_definition_id

STRING

模型定义的 ID

model_definition

GatewayModelDefinition

完整的模型定义(通过 JOIN 填充)

weight

FLOAT

用于流量分配的回退权重

created_at

INT64

创建映射时的时间戳(自纪元以来的毫秒数)

created_by

STRING

创建映射的用户 ID

linkage_type

GatewayModelLinkageType

链接类型

fallback_order

INT32

回退尝试的顺序(仅适用于 FALLBACK 链接,PRIMARY 为 NULL)

GatewayEndpointTag

与端点关联的标签

字段名称

类型

描述

key

STRING

标签键

value

STRING

标签值

GatewayModelDefinition

可跨端点共享的可重用模型定义

字段名称

类型

描述

model_definition_id

STRING

此模型定义的唯一标识符

name

STRING

用于识别和重用的用户友好名称

secret_id

STRING

包含身份验证凭据的密钥的 ID

secret_name

STRING

用于显示目的的密钥名称

provider

STRING

LLM 提供商(例如,“openai”、“anthropic”、“cohere”、“bedrock”)

model_name

STRING

提供商特定的模型标识符(例如,“gpt-4o”、“claude-3-5-sonnet”)

created_at

INT64

创建模型定义时的时间戳(自纪元以来的毫秒数)

last_updated_at

INT64

最后更新模型定义时的时间戳(自纪元以来的毫秒数)

created_by

STRING

创建模型定义的用户 ID

last_updated_by

STRING

最后更新模型定义的用户 ID

GatewaySecretInfo

密钥元数据实体(不包含解密的密钥值)

字段名称

类型

描述

secret_id

STRING

密钥的唯一标识符(UUID)

secret_name

STRING

密钥的友好名称(必须唯一)

masked_values

一个 MaskedValuesEntry 数组

密钥值的屏蔽版本,用于显示为键值对。对于简单的 API 密钥:{“api_key”: “sk-…xyz123”} 对于复合凭据:{"aws_access_key_id": "AKI...1234", "aws_secret_access_key": "***"}

created_at

INT64

创建密钥时的时间戳(自纪元以来的毫秒数)

last_updated_at

INT64

最后更新密钥时的时间戳(自纪元以来的毫秒数)

provider

STRING

LLM 提供商标识符(例如,“openai”、“anthropic”、“cohere”)

created_by

STRING

创建密钥的用户 ID

last_updated_by

STRING

最后更新密钥的用户 ID

auth_config

一个 AuthConfigEntry 数组

特定于提供商的身份验证配置(例如,auth_mode、region、project_id)

InputTag

输入的标签。

字段名称

类型

描述

key

STRING

标签键。此字段是必需的。

value

STRING

标签值。此字段是必需的。

MaskedValuesEntry

字段名称

类型

描述

key

STRING

value

STRING

Metric

与运行关联的指标,表示为键值对。

字段名称

类型

描述

key

STRING

标识此指标的键。

value

DOUBLE

与此指标关联的值。

timestamp

INT64

记录此指标的时间戳。

step

INT64

记录指标的步数。

ModelInput

代表 Run 的 LoggedModel 或 Registered Model Version 输入。

字段名称

类型

描述

model_id

STRING

模型的唯一标识符。此字段是必需的。

ModelMetric

与模型关联的指标,表示为键值对。从 MLflow 指标复制

字段名称

类型

描述

key

STRING

标识此指标的键。

value

DOUBLE

与此指标关联的值。

timestamp

INT64

记录此指标的时间戳。

step

INT64

记录指标的步数。

ModelOutput

代表 Run 的 LoggedModel 输出。

字段名称

类型

描述

model_id

STRING

模型的唯一标识符。此字段是必需的。

step

INT64

生成模型的步数。此字段是必需的。

ModelParam

模型版本的参数。

字段名称

类型

描述

name

STRING

参数的名称。

value

STRING

与名称关联的参数值,可以为空

ModelVersion

字段名称

类型

描述

name

STRING

模型的唯一名称

version

STRING

模型的版本号。

creation_timestamp

INT64

创建此 model_version 时记录的时间戳。

last_updated_timestamp

INT64

最后一次更新此 model_version 元数据时记录的时间戳。

user_id

STRING

创建此 model_version 的用户。

current_stage

STRING

model_version 的当前阶段。

description

STRING

model_version 的描述。

source

STRING

指示源模型工件位置的 URI,在创建 model_version 时使用

run_id

STRING

创建 model_version 时使用的 MLflow 运行 ID,如果 source 是由存储在 MLflow 跟踪服务器中的实验运行生成的。

status

ModelVersionStatus

model_version 的当前状态

status_message

STRING

有关当前 status 的详细信息,如果处于待定或失败状态。

tags

一个 ModelVersionTag 实体的数组

标签:此 model_version 的额外元数据键值对。

run_link

STRING

运行链接:指向生成此版本的运行的直接链接。仅当源运行来自与注册表服务器不同的跟踪服务器时,此字段才在模型版本创建时设置。

aliases

一个 STRING 数组

指向此 model_version 的别名。

model_id

STRING

模型版本的可选 model_id,用于将已注册模型链接到源已记录的模型

model_params

一个 ModelParam 数组

模型的可选参数。

model_metrics

一个 ModelMetric 数组

模型的可选指标。

deployment_job_state

ModelVersionDeploymentJobState

此模型版本的部署作业状态

ModelVersionDeploymentJobState

字段名称

类型

描述

job_id

STRING

run_id

STRING

job_state

State

run_state

DeploymentJobRunState

current_task_name

STRING

ModelVersionTag

模型版本的标签。

字段名称

类型

描述

key

STRING

标签键。

value

STRING

标签值。

Param

与运行关联的参数。

字段名称

类型

描述

key

STRING

标识此参数的键。

value

STRING

与此参数关联的值。

RegisteredModel

字段名称

类型

描述

name

STRING

模型的唯一名称。

creation_timestamp

INT64

创建此 registered_model 时记录的时间戳。

last_updated_timestamp

INT64

最后一次更新此 registered_model 元数据时记录的时间戳。

user_id

STRING

创建此 registered_model 的用户 注意:此字段当前未返回。

description

STRING

registered_model 的描述。

latest_versions

一个 ModelVersion 实体的数组

每个阶段的最新模型版本的集合。仅包含当前状态为 READY 的模型。

tags

一个 RegisteredModelTag 实体的数组

标签:此 registered_model 的额外元数据键值对。

aliases

一个 RegisteredModelAlias 数组

指向与此 registered_model 关联的模型版本的别名。

deployment_job_id

STRING

此模型的部署作业 ID。

deployment_job_state

State

此模型的部署作业状态。

RegisteredModelAlias

已注册模型的别名

字段名称

类型

描述

alias

STRING

别名的名称。

version

STRING

别名指向的模型版本号。

RegisteredModelTag

已注册模型的标签

字段名称

类型

描述

key

STRING

标签键。

value

STRING

标签值。

Run

单个运行。

字段名称

类型

描述

info

RunInfo

运行元数据。

data

RunData

运行数据。

inputs

RunInputs

运行输入。

输出

RunOutputs

运行输出。

RunData

运行数据(指标、参数和标签)。

字段名称

类型

描述

metrics

一个 Metric 数组

运行指标。

params

一个 Param 数组

运行参数。

tags

一个 RunTag 数组

额外的元数据键值对。

RunInfo

单个运行的元数据。

字段名称

类型

描述

run_id

STRING

运行的唯一标识符。

run_uuid

STRING

[已弃用,请使用 run_id] 运行的唯一标识符。此字段将在未来的 MLflow 版本中删除。

run_name

STRING

运行的名称。

experiment_id

STRING

实验 ID。

user_id

STRING

启动运行的用户。此字段自 MLflow 1.0 起已弃用,并将在未来的 MLflow 版本中删除。请改用“mlflow.user”标签。

status

RunStatus

运行的当前状态。

start_time

INT64

运行开始时间的 Unix 时间戳(毫秒)。

end_time

INT64

运行结束时间的 Unix 时间戳(毫秒)。

artifact_uri

STRING

应上传工件的目录 URI。这可以是本地路径(以“/”开头),也可以是分布式文件系统 (DFS) 路径,例如 s3://bucket/directorydbfs:/my/directory。如果未设置,则选择本地 ./mlruns 目录。

lifecycle_stage

STRING

实验当前的生命周期阶段:OneOf(“active”, “deleted”)

RunInputs

运行输入。

字段名称

类型

描述

dataset_inputs

一个 DatasetInput 数组

Run 的数据集输入。

model_inputs

一个 ModelInput 数组

Run 的模型输入。

RunOutputs

Run 的输出。

字段名称

类型

描述

model_outputs

一个 ModelOutput 数组

Run 的模型输出。

RunTag

运行的标签。

字段名称

类型

描述

key

STRING

标签键。

value

STRING

标签值。

Scorer

代表数据库中评分实体的评分器。

字段名称

类型

描述

experiment_id

INT32

实验 ID。

scorer_name

STRING

评分器名称。

scorer_version

INT32

评分器的版本。

serialized_scorer

STRING

序列化的评分器字符串。

creation_time

INT64

评分器版本的创建时间(自纪元以来的毫秒数)。

scorer_id

STRING

评分器的唯一标识符。

SecretValueEntry

字段名称

类型

描述

key

STRING

value

STRING

SecretValueEntry

字段名称

类型

描述

key

STRING

value

STRING

DeploymentJobRunState

Name

描述

DEPLOYMENT_JOB_RUN_STATE_UNSPECIFIED

未找到有效的部署作业

运行中

成功

失败

待定

批准

FallbackStrategy

用于路由的回退策略(为额外的策略提供面向未来的兼容性)

Name

描述

SEQUENTIAL

顺序回退:按指定顺序尝试模型

GatewayModelLinkageType

端点与模型定义之间的链接类型

Name

描述

PRIMARY

主要链接:用于路由流量

FALLBACK

回退链接:用于故障转移

ModelVersionStatus

Name

描述

PENDING_REGISTRATION

注册新模型版本的请求处于待定状态,因为服务器正在执行后台任务。

FAILED_REGISTRATION

注册新模型版本的请求失败。

READY

模型版本已准备好使用。

RoutingStrategy

端点的路由策略

Name

描述

REQUEST_BASED_TRAFFIC_SPLIT

基于请求的流量拆分:根据权重分配流量

RunStatus

运行的状态。

Name

描述

运行中

Run 已启动。

SCHEDULED

Run 计划在稍后时间运行。

FINISHED

Run 已完成。

失败

Run 执行失败。

KILLED

Run 被用户杀死。

State

Name

描述

DEPLOYMENT_JOB_CONNECTION_STATE_UNSPECIFIED

未设置

默认状态

CONNECTED

已连接的作业:作业存在,所有者具有 ACL,并且存在必需的作业参数

未找到

作业被删除或所有者的作业 ACL 被移除

REQUIRED_PARAMETERS_CHANGED

必需的作业参数已更改

ViewType

ListExperiments 查询的视图类型。

Name

描述

ACTIVE_ONLY

默认值。仅返回活动的实验。

DELETED_ONLY

仅返回已删除的实验。

ALL

获取所有实验。