Поиск правильного формата 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."}