Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 25 additions & 0 deletions openeo/processes.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
"""
Expand Down Expand Up @@ -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:
Expand Down