Changeset 13707 for trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/GBM

Timestamp:

03/15/16 17:36:38 (9 years ago)

Author:

gkronber

Message:

#1795: fixed a problem for datasets with variables that don't have double type

File:

• trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/GBM/GradientBoostingRegressionAlgorithm.cs (modified) (1 diff)

trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/GBM/GradientBoostingRegressionAlgorithm.cs

@@ -231 +231 @@
       var problemData = Problem.ProblemData;
       var targetVarName = Problem.ProblemData.TargetVariable;
+      var activeVariables = problemData.AllowedInputVariables.Concat(new string[] { problemData.TargetVariable });
       var modifiableDataset = new ModifiableDataset(
-        problemData.Dataset.VariableNames,
-        problemData.Dataset.VariableNames.Select(v => problemData.Dataset.GetDoubleValues(v).ToList()));
+        activeVariables,
+        activeVariables.Select(v => problemData.Dataset.GetDoubleValues(v).ToList()));
 
       var trainingRows = problemData.TrainingIndices;