Changeset 12063

Timestamp:

02/23/15 18:01:58 (9 years ago)

Author:

ehopf

Message:

#2335: Added code to keep the user defined positive class in the generated problem data of a classification problem. (Defect 2 fix)

File:

• branches/DataPreprocessingImprovements/HeuristicLab.DataPreprocessing/3.4/ProblemDataCreator.cs (modified) (1 diff)

branches/DataPreprocessingImprovements/HeuristicLab.DataPreprocessing/3.4/ProblemDataCreator.cs

@@ -77 +77 @@
 
     private IDataAnalysisProblemData CreateClassificationData(ClassificationProblemData oldProblemData) {
+      // target variable must be double and must exist in the new dataset
       var targetVariable = oldProblemData.TargetVariable;
-      // target variable must be double and must exist in the new dataset
-      return new ClassificationProblemData(ExportedDataset, GetDoubleInputVariables(targetVariable), targetVariable, Transformations);
+      var newProblemData = new ClassificationProblemData(ExportedDataset, GetDoubleInputVariables(targetVariable), targetVariable, Transformations);
+      newProblemData.PositiveClass = oldProblemData.PositiveClass;
+      return newProblemData;
     }