Changeset 11382 for trunk/sources/HeuristicLab.DataPreprocessing

Timestamp:

09/18/14 19:23:10 (10 years ago)

Author:

bburlacu

Message:

#2246: Preserve selected variables when creating new problem data.

File:

• trunk/sources/HeuristicLab.DataPreprocessing/3.4/ProblemDataCreator.cs (modified) (2 diffs)

trunk/sources/HeuristicLab.DataPreprocessing/3.4/ProblemDataCreator.cs

@@ -22 +22 @@
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using HeuristicLab.Common;
 using HeuristicLab.Problems.DataAnalysis;
@@ -58 +59 @@
 
       SetTrainingAndTestPartition(problemData);
+      // set the input variables to the correct checked state
+      var inputVariables = problemData.InputVariables.ToDictionary(x => x.Value, x => x);
+      foreach (var variable in oldProblemData.InputVariables) {
+        bool @checked = oldProblemData.InputVariables.ItemChecked(variable);
+        problemData.InputVariables.SetItemCheckedState(inputVariables[variable.Value], @checked);
+      }
 
       return problemData;