Получение исключения при попытке получить оценки из размещенной модели формата tf savemodelbundle с использованием Java
Я получаю исключение, пытаясь получить оценки из размещенной модели формата savemodelbundle. Я использую GRPC версии 1.1.2.
Исключение составляет
io.grpc.StatusRuntimeException: INTERNAL: tensorflow.serving.PredictionService/Predict: INTERNAL: Expected prediction signature method_name to be one of {tensorflow/serving/predict, tensorflow/serving/classify, tensorflow/serving/regress}. Was:
at io.grpc.stub.ClientCalls.toStatusRuntimeException(ClientCalls.java:227)
at io.grpc.stub.ClientCalls.getUnchecked(ClientCalls.java:208)
at io.grpc.stub.ClientCalls.blockingUnaryCall(ClientCalls.java:141)
at tensorflow.serving.PredictionServiceGrpc$PredictionServiceBlockingStub.predict(PredictionServiceGrpc.java:287)
Мой запрос выглядит так:
ModelSpec theModelSpec = ModelSpec.newBuilder().setName("export").setSignatureName("serving_default").build();
tensorflow.serving.Predict.PredictRequest.Builder theRequestBuilder = PredictRequest.newBuilder().setModelSpec(theModelSpec).putInputs("input", createTensorProto(values));
Я знаю, что мне не хватает какого-то параметра. Любые выводы будут полезны. Спасибо!!