diff --git a/openeo/processes.py b/openeo/processes.py index fcc13312f..7226b948b 100644 --- a/openeo/processes.py +++ b/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: