google-cloud-ml - Cloud ML 批量预测目前支持哪些数据格式？

Question

我在训练有素的模型中构建了推理图，并希望使用批量预测来预测许多记录。如何在输入文件中指定输入？

Answer 1

Cloud ML 目前支持三种数据格式。一个是文本文件，其中的每一行都是您要预测的记录。第二种和第三种格式是TFRecords；支持压缩和 gzip 压缩。TFRecord 文件是一个存储字节的容器，通常是二进制数据，例如序列化的示例原型。这些字节被直接馈送到预测图中。您必须在请求的 data_format 字段（TEXT、TF_RECORD、TF_RECORD_GZIP）中指定它们。

对于文本格式，每一行都是 JSON 对象或 UTF8 字符串。在前者的情况下，键是输入张量名称，值是将输入推理图中的数据。如果您的图表只有一个输入张量，您可以跳过 JSON 并只保存换行符分隔的字符串。

这里有些例子：

你有四个输入张量，即索引、高度、名称和图像

{“index”: 100, “height”: 5.5, “name”: “Alice”, “image”: [0.0, 0.0, 0.123, 0.17,0,0]}
{“index”: 101, “height”: 5.8, “name”: “John”, “image”: [0.0, 0.21, 0.09, 0.5, 0,0]}
...

你有一个字符串输入张量。无需指定名称。

“This is a string input”
“That is another string input”
...

你有一个标量类型的张量。无需指定名称。

1445
425
3412
...

你有一个输入张量，它是一个 numpy 数组。无需指定名称。

[0, 3.14, 2.718, 0.0, 1.414]
[1.618, 299.7, 8.314, 0.0, 0.0]
...

请注意，多张量输入案例中的名称必须与推理图中输入集合中定义的别名匹配。