From d97ac5407a082d0cff3a04b0a51b35fe65af58bd Mon Sep 17 00:00:00 2001
From: Dimitris Athanasiou <dimitris@elastic.co>
Date: Tue, 17 Nov 2020 12:48:25 +0200
Subject: [PATCH] [7.x][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.

Backport of #83151
---
 .../overrides/ml.evaluate_data_frame.json     | 86 ++++++++++++++++---
 1 file changed, 73 insertions(+), 13 deletions(-)

diff --git a/x-pack/plugins/console_extensions/server/lib/spec_definitions/json/overrides/ml.evaluate_data_frame.json b/x-pack/plugins/console_extensions/server/lib/spec_definitions/json/overrides/ml.evaluate_data_frame.json
index bf78cf7f09aaf..f6cc8bf2693ce 100644
--- a/x-pack/plugins/console_extensions/server/lib/spec_definitions/json/overrides/ml.evaluate_data_frame.json
+++ b/x-pack/plugins/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
+                }
+              }
+            }
           }
-        }
+        ]
       }
     }
   }