Changeset 13066 for trunk/sources/HeuristicLab.Tests

Timestamp:

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

Author:

gkronber

Message:

#2450: adapted unit test

File:

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

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;