Changeset 17833

Timestamp:

02/09/21 17:08:22 (4 years ago)

Author:

bburlacu

Message:

#2806: Set positive class value when cloning problem data.

File:

• branches/3087_Ceres_Integration/HeuristicLab.Algorithms.DataAnalysis/3.4/CrossValidation.cs (modified) (1 diff)

branches/3087_Ceres_Integration/HeuristicLab.Algorithms.DataAnalysis/3.4/CrossValidation.cs

@@ -515 +515 @@
           dataset = dataset.Shuffle(random);
         }
-        var problemDataClone = new ClassificationProblemData(dataset, Problem.ProblemData.AllowedInputVariables, targetVariable);
+        var problemData = (IClassificationProblemData)Problem.ProblemData;
+        var problemDataClone = new ClassificationProblemData(dataset, problemData.AllowedInputVariables, targetVariable);
         // set partitions of problem data clone correctly
         problemDataClone.TrainingPartition.Start = SamplesStart.Value; problemDataClone.TrainingPartition.End = SamplesEnd.Value;
         problemDataClone.TestPartition.Start = SamplesStart.Value; problemDataClone.TestPartition.End = SamplesEnd.Value;
+        problemDataClone.PositiveClass = problemData.PositiveClass;
         // clone models
         var ensembleSolution = new ClassificationEnsembleSolution(problemDataClone);