Changeset 13066

Timestamp:

10/27/15 08:49:57 (9 years ago)

Author:

gkronber

Message:

#2450: adapted unit test

Location:

trunk/sources

Files:

• HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesModelSurrogate.cs (modified) (1 diff)
• HeuristicLab.Tests/HeuristicLab.Algorithms.DataAnalysis-3.4/GradientBoostingTest.cs (modified) (1 diff)

trunk/sources/HeuristicLab.Algorithms.DataAnalysis/3.4/GradientBoostedTrees/GradientBoostedTreesModelSurrogate.cs

@@ -36 +36 @@
     // don't store the actual model!
     private IRegressionModel actualModel; // the actual model is only recalculated when necessary
+    public IRegressionModel Model { get { return actualModel; } }
 
     [Storable]

trunk/sources/HeuristicLab.Tests/HeuristicLab.Algorithms.DataAnalysis-3.4/GradientBoostingTest.cs

@@ -269 +269 @@
       problemData.TestPartition.End = nRows;
       var solution = GradientBoostedTreesAlgorithmStatic.TrainGbm(problemData, new SquaredErrorLoss(), maxSize, nu: 1, r: 1, m: 1, maxIterations: 1, randSeed: 31415);
-      var model = (GradientBoostedTreesModel)solution.Model;
+      var model = (GradientBoostedTreesModel)((GradientBoostedTreesModelSurrogate)solution.Model).Model;
       var treeM = model.Models.Skip(1).First() as RegressionTreeModel;