为Google AI平台推断找到正确的JSON格式

时间:2019-11-25 11:04:04

标签: google-cloud-platform gcloud google-cloud-ml tf-hub

我正在尝试使用gcloudsi-ai-platform本地预测命令在Google Universal Sentence Encoder上运行预测。我的命令如下所示:

gcloud ai-platform local predict --model-dir=/Users/x/Downloads/universal-sentence-encoder/ --json-instances=instances.json --verbosity debug

instances.json看起来像这样:

{"inputs": ["Hello World."]}

我从gcloud获得了以下内容:

cloud.ml.prediction.prediction_utils.PredictionError: Failed to run the provided model: Exception during running the graph: Cannot feed value of shape (1, 1) for Tensor 'serving_default_inputs:0', which has shape '(?,)' (Error code: 2)

我认为我的输入格式错误,但是我找不到正确的格式。有谁知道如何检查保存的模型以找出正确的输入格式?

2 个答案:

答案 0 :(得分:1)

Svetlana's answer 是正确的。您可以通过重复该行来批处理多个请求,例如:

{"inputs": "Hello World."}
{"inputs": "Hello Mars."}

答案 1 :(得分:0)

您可能需要提供字符串而不是列表。以下输入可能有效:

{"inputs": "Hello World."}