Databricks ML 模型权限

Question

全部，

从模型菜单中，我从各个版本中选择了一个模型（贷款审批数据），并能够将其设置为“实时推理”。我对 1994 年人口普查数据也做了同样的事情。我有两个有关使用的问题：

1. 对于贷款审批模型，当我单击“使用”查询端点时，我没有看到示例请求。但是，对于 1994 年人口普查模型，我能够看到示例请求和响应。有理由吗？我的意思是为什么有些模型会生成示例请求而有些则不会？

2. 如何授予对模型的访问权限？我必须创建一个 PAT 令牌并将其用作邮递员中的不记名令牌，才能使 1994 年人口普查模型发挥作用。模型及其端点需要哪些权限才能正常运行？

谢谢， 戈皮

Answer 1

通常，当您使用使用自定义训练模型创建的端点时，您将不会收到示例请求。

只有内置的查询端点时才会有示例请求。

也有可能取决于模型签名

以及本文档中提到的输入示例

因此，每当您记录和注册模型时，您都会连同签名和输入示例一起进行操作。

示例代码

from mlflow.models.signature import infer_signature

model_path = "models/diabetes_rf"
signature = infer_signature(X_train, predictions)
input_example = X_train[:5]
mlflow.sklearn.log_model(rf, model_path, signature=signature, input_example=input_example)
model_uri = f"runs:/{mlflow.active_run().info.run_id}/{model_path}"
model_details = mlflow.register_model(model_uri, "RandomForestDiabetesModel")

接下来，当涉及到模型的权限时，您可以在模型选项卡中处理它，如图所示。

在这里，您可以设置谁可以访问模型并使用它。

要使用服务端点，您只需要 PAT 令牌即可成功请求。