我可以在 Python 中使用导出的 AutoML Tables 模型吗？

问题描述

我可以从 AutoML 表中导出模型并在本地与 tensorflow 一起使用吗？

我发现您可以以多种格式导出某些 Vision 模型，但对于表格，我只能将 tf 边缘模型导出到 Cloud Storage。那么我可以以某种方式在本地运行它来执行预测吗？或者是否有另一种方法可以将模型导出/转换为可在本地使用的格式？

解决方法

AutoML 表可以导出 tf 保存的模型（.pb 格式）。您可以通过使用 curl 发送请求来做到这一点。见Exporting Models in AutoML Tables 有关如何使用 models.export 的详细说明。

注意：用于测试的导出模型来自 AutoML Tables quickstart。

JSON 请求（使用 tf_saved_model 以 SavedModel 格式导出 tensorflow 模型）：

{
  "outputConfig": {
    "modelFormat": "tf_saved_model","gcsDestination": {
      "outputUriPrefix": "your-gcs-destination"
    }
  }
}

curl 命令（假设模型在 us-central1 中）：

curl -X POST \
-H "Authorization: Bearer "$(gcloud auth application-default print-access-token) \
-H "Content-Type: application/json; charset=utf-8" \
-d @request.json \
https://automl.googleapis.com/v1beta1/projects/your-project-id/locations/us-central1/models/your-model-id:export

导出模型：