Merge pull request #830 from Open-EO/issue525-onnx-process

issue525 add onnx process

Author: ElienVandermaesenVITO

openeo/processes.py

@@ -2133,6 +2133,18 @@ def predict_curve(self, function, dimension, labels=UNSET) -> ProcessBuilder:
             labels=labels
         )
 
+    @openeo_process
+    def predict_onnx(self, model) -> ProcessBuilder:
+        """
+        Predict values based on an onnx model
+
+        :param self: An array of numbers.
+        :param model: an url to an onnx model
+
+        :return: The predicted value.
+        """
+        return predict_onnx(data=self, model=model)
+
     @openeo_process
     def predict_random_forest(self, model) -> ProcessBuilder:
         """
@@ -4910,6 +4922,19 @@ def predict_curve(parameters, function, dimension, labels=UNSET) -> ProcessBuild
         labels=labels
     )
 
+@openeo_process
+def predict_onnx(data, model) -> ProcessBuilder:
+    """
+    Predict values based on a Random Forest model
+
+    :param data: An array of numbers.
+    :param model: A model object that can be trained with the processes ``fit_regr_random_forest()``
+        (regression) and ``fit_class_random_forest()`` (classification).
+
+    :return: The predicted value. Returns `null` if any of the given values in the array is a no-data value.
+    """
+    return _process("predict_onnx", data=data, model=model)
+
 
 @openeo_process
 def predict_random_forest(data, model) -> ProcessBuilder: