当部署到顶点ai时，如何将阈值传递给BigQueryML模型

Question

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

已部署的模型使用0.5的阈值。

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

Answer 1

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

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

根据文档，在线预测的输出包含置信度分值，客户端需要在代码中对其进行解释。您可以通过阈值来过滤预测结果。