[ML] Fix default selected fields.

elastic · Mar 30, 2020 · 4cc34ac · 4cc34ac
1 parent 4956c29
commit 4cc34ac
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/x-pack/plugins/ml/public/application/data_frame_analytics/common/fields.ts b/x-pack/plugins/ml/public/application/data_frame_analytics/common/fields.ts
@@ -286,9 +286,11 @@ export const getDefaultFieldsFromJobCaps = (
     ...featureImportanceFields,
   ].sort(({ name: a }, { name: b }) => sortRegressionResultsFields(a, b, jobConfig));
 
-  let selectedFields = allFields
-    .slice(0, DEFAULT_REGRESSION_COLUMNS * 2)
-    .filter((field: any) => field.name === predictedField || !field.name.includes('.keyword'));
+  let selectedFields = allFields.filter(
+    (field: any) =>
+      field.name === predictedField ||
+      (!field.name.includes('.keyword') && !field.name.includes('.feature_importance.'))
+  );
 
   if (selectedFields.length > DEFAULT_REGRESSION_COLUMNS) {
     selectedFields = selectedFields.slice(0, DEFAULT_REGRESSION_COLUMNS);