部署到顶点 ai 时如何将阈值传递给 BigQuery ML model

Question

我将 BigQuery ML model 导出到云存储，然后将生成的 tensorflow model 导入到顶点 ai。

部署的 model 使用阈值 0.5。

如果我在 BigQuery 中进行预测，我会使用 ML.predict 和 STRUCT（0.05 作为 THRESHOLD）。 我如何告诉 Vertex AI 类似地将 THRESHOLD 传递给所有输入请求？

Answer 1

根据 BigQuery ML 文档，阈值仅适用于二元分类模型。

将导出的 model 部署到 Vertex AI 时，您可以按照此处所述继续从中获取在线预测。

根据文档，在线预测的 output 包含置信度得分值，需要由客户端在您的代码中解释。 您可以按阈值过滤预测结果。