Changeset 12869 for branches/crossvalidation-2434/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesAlgorithm.cs
- Timestamp:
- 08/17/15 16:35:47 (9 years ago)
- Location:
- branches/crossvalidation-2434
- Files:
-
- 3 edited
Legend:
- Unmodified
- Added
- Removed
-
branches/crossvalidation-2434
- Property svn:mergeinfo changed
/trunk/sources merged: 12835-12837,12839,12844-12846,12851,12855,12868
- Property svn:mergeinfo changed
-
branches/crossvalidation-2434/HeuristicLab.Algorithms.DataAnalysis
- Property svn:mergeinfo changed
/trunk/sources/HeuristicLab.Algorithms.DataAnalysis merged: 12868
- Property svn:mergeinfo changed
-
branches/crossvalidation-2434/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesAlgorithm.cs
r12632 r12869 233 233 // produce solution 234 234 if (CreateSolution) { 235 var surrogateModel = new GradientBoostedTreesModelSurrogate(problemData, (uint)Seed, lossFunction.ToString(), 236 Iterations, MaxSize, R, M, Nu, state.GetModel()); 237 235 238 // for logistic regression we produce a classification solution 236 239 if (lossFunction is LogisticRegressionLoss) { 237 var model = new DiscriminantFunctionClassificationModel(s tate.GetModel(),240 var model = new DiscriminantFunctionClassificationModel(surrogateModel, 238 241 new AccuracyMaximizationThresholdCalculator()); 239 242 var classificationProblemData = new ClassificationProblemData(problemData.Dataset, … … 245 248 } else { 246 249 // otherwise we produce a regression solution 247 Results.Add(new Result("Solution", new RegressionSolution(s tate.GetModel(), problemData)));250 Results.Add(new Result("Solution", new RegressionSolution(surrogateModel, problemData))); 248 251 } 249 252 }
Note: See TracChangeset
for help on using the changeset viewer.