Method main

serving/sdk/java/src/main/java/Demo.java:38–90

(String[] args)

    }

    public static void main(String[] args) {
        Demo demo = new Demo();
        String modelEntry = "";
        int[] state = {0};
        Pointer model = Processor.INSTANCE.initialize(modelEntry, modelConfig, state);
        if (state[0] == -1) {
            System.err.println("initialize error");
        }

        // input type: float
        Predict.ArrayDataType dtype = Predict.ArrayDataType.DT_FLOAT;
        // input shape: [1, 1]
        Predict.ArrayShape arrayShape =
            Predict.ArrayShape.newBuilder()
            .addDim(1)
            .addDim(1)
            .build();
        // input array
        Predict.ArrayProto input =
            Predict.ArrayProto.newBuilder()
            .addFloatVal((float) 1.0)
            .setDtype(dtype)
            .setArrayShape(arrayShape)
            .build();
        // PredictRequest
        Predict.PredictRequest req =
            Predict.PredictRequest.newBuilder()
            .setSignatureName("serving_default")
            .addOutputFilter("y:0")
            .putInputs("x:0", input)
            .build();
        byte[] buffer = req.toByteArray();
        int size = req.getSerializedSize();

        // do process
        PointerByReference output = new PointerByReference();
        int[] outputSize = {0};
        state[0] = Processor.INSTANCE.process(model, buffer, size, output, outputSize);

        // parse response
        byte[] outputString = output.getValue().getByteArray(0, outputSize[0]);
        String s = new String(outputString);
        try {
            Predict.PredictResponse resp =
                Predict.PredictResponse.newBuilder()
                .mergeFrom(outputString)
                .build();
            System.out.println(resp.toString());
        } catch (Exception e) {
            System.err.println("parse response error");
        }

    }
}