Поиск правильного формата JSON для вывода Google AI Platform

Я пытаюсь выполнять прогнозы на универсальном кодировщике предложений Googles с помощью команды gclouds ai-platform local predic. Моя команда выглядит так:

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

А instance.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 ответа

Решение

Светлана ответила правильно. Вы можете группировать несколько запросов, повторяя строку, например:

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

Вероятно, вам нужно указать строку вместо списка. Следующий ввод может работать:

{"inputs": "Hello World."}
Другие вопросы по тегам