微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决我可以在 Python 中使用导出的 AutoML Tables 模型吗?

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

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

enter image description here

解决方法

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

导出模型: enter image description here

enter image description here

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。