[ML] Update console autocomplete for ML data frame evaluate API (#83151)

We have added evaluation for all types of data frame analysis since the last update. This commit updates autocomplete accordingly.
elastic · Nov 17, 2020 · 95ff10b · 95ff10b
1 parent 95e44f2
commit 95ff10b
Showing 1 changed file with 73 additions and 13 deletions.
diff --git a/...console_extensions/server/lib/spec_definitions/json/overrides/ml.evaluate_data_frame.json b/...console_extensions/server/lib/spec_definitions/json/overrides/ml.evaluate_data_frame.json
@@ -1,23 +1,83 @@
 {
   "ml.evaluate_data_frame": {
     "data_autocomplete_rules": {
+      "index": "",
       "evaluation": {
-        "actual_field": "FIELD_NAME",
-        "predicted_probability_field": "FIELD_NAME",
-        "metrics": {
-          "auc_roc": {
-            "include_curve": true
+        "__one_of": [
+          {
+            "outlier_detection": {
+              "__template": {
+                "actual_field": "FIELD_NAME",
+                "predicted_probability_field": "FIELD_NAME"
+              },
+              "actual_field": "FIELD_NAME",
+              "predicted_probability_field": "FIELD_NAME",
+              "metrics": {
+                "auc_roc": {
+                  "include_curve": false
+                },
+                "precision": {
+                  "at": []
+                },
+                "recall": {
+                  "at": []
+                },
+                "confusion_matrix": {
+                  "at": []
+                }
+              }
+            }
           },
-          "precision": {
-            "at": []
+          {
+            "regression": {
+              "__template": {
+                "actual_field": "FIELD_NAME",
+                "predicted_field": "FIELD_NAME"
+              },
+              "actual_field": "FIELD_NAME",
+              "predicted_field": "FIELD_NAME",
+              "metrics": {
+                "mse": {},
+                "msle": {
+                  "offset": 1.0
+                },
+                "r_squared": {},
+                "huber": {
+                  "delta": 1.0
+                }
+              }
+            }
           },
-          "recall": {
-            "at": []
-          },
-          "confusion_matrix": {
-            "at": []
+          {
+            "classification": {
+              "__template": {
+                "actual_field": "FIELD_NAME",
+                "predicted_field": "FIELD_NAME",
+                "top_classes_field": "FIELD_NAME"
+              },
+              "actual_field": "FIELD_NAME",
+              "predicted_field": "FIELD_NAME",
+              "top_classes_field": "FIELD_NAME",
+              "metrics": {
+                "accuracy": {},
+                "precision": {},
+                "recall": {},
+                "multiclass_confusion_matrix": {
+                  "size": 10
+                },
+                "precision": {},
+                "recall": {},
+                "auc_roc": {
+                  "__template": {
+                    "class_name": ""
+                  },
+                  "class_name": "",
+                  "include_curve": false
+                }
+              }
+            }
           }
-        }
+        ]
       }
     }
   }