rename predict_catboost_probabilities process to predict_probabilities

see Open-EO/openeo-geotrellis-extensions#48
Open-EO · May 11, 2022 · 9027037 · 9027037
1 parent 836d9d9
commit 9027037
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/openeo_driver/ProcessGraphDeserializer.py b/openeo_driver/ProcessGraphDeserializer.py
@@ -761,8 +761,8 @@ def predict_random_forest(args: dict, env: EvalEnv) -> SaveResult:
 def predict_catboost(args: dict, env: EvalEnv) -> SaveResult:
     pass
 
-@process_registry_100.add_function(spec=read_spec("openeo-processes/experimental/predict_catboost_probabilities.json"))
-def predict_catboost_probabilities(args: dict, env: EvalEnv) -> SaveResult:
+@process_registry_100.add_function(spec=read_spec("openeo-processes/experimental/predict_probabilities.json"))
+def predict_probabilities(args: dict, env: EvalEnv) -> SaveResult:
     pass
 
 @process

diff --git a/...ental/predict_catboost_probabilities.json → ...s/experimental/predict_probabilities.json b/...ental/predict_catboost_probabilities.json → ...s/experimental/predict_probabilities.json
@@ -1,7 +1,7 @@
 {
-    "id": "predict_catboost_probabilities",
-    "summary": "Predict class probabilities from a Catboost model",
-    "description": "Applies a Catboost machine learning model to an array and predicts a probability for each class.",
+    "id": "predict_probabilities",
+    "summary": "Predict class probabilities from a machine learning model",
+    "description": "Applies a machine learning model to an array and predicts a probability for each class.",
     "categories": [
         "machine learning",
         "reducer"
@@ -23,7 +23,7 @@
         },
         {
             "name": "model",
-            "description": "A model object that can be trained with the processes ``fit_regr_catboost()`` (regression) and ``fit_class_catboost()`` (classification).",
+            "description": "A model object that can be trained with the processes ``fit_regr_<model_name>()`` (regression) and ``fit_class_<model_name>()`` (classification).",
             "schema": {
                 "type": "object",
                 "subtype": "ml-model"