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]
...

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