Orca openvino estimator (intel-analytics#3011)

* init * add distripredict * javardd * java spark context * test * fix * remove tes * remove * add type check * meet comments * fix import * fix * fix style * combine testcase * fix style * replaceable variable
yangw1234 · Sep 23, 2021 · 2bfe5e0 · 2bfe5e0
1 parent 3f9026b
commit 2bfe5e0
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/python/orca/src/bigdl/orca/inference/inference_model.py b/python/orca/src/bigdl/orca/inference/inference_model.py
@@ -146,3 +146,18 @@ def predict(self, inputs):
                              jinputs,
                              input_is_table)
         return KerasNet.convert_output(output)
+
+    def distributed_predict(self, inputs, sc):
+        data_type = inputs.map(lambda x: x.__class__.__name__).first()
+        input_is_table = False
+        if data_type == "list":
+            input_is_table = True
+        jinputs = inputs.map(lambda x: Layer.check_input(x)[0])
+
+        output = callZooFunc(self.bigdl_type,
+                             "inferenceModelDistriPredict",
+                             self.value,
+                             sc,
+                             jinputs,
+                             input_is_table)
+        return output.map(lambda x: KerasNet.convert_output(x))