Changeset 12661 for branches/GBT-trunkintegration/Tests/Test.cs

Timestamp:

07/07/15 17:58:01 (9 years ago)

Author:

gkronber

Message:

#2261: regression tree builder should not be used from outside (made internal)

File:

• branches/GBT-trunkintegration/Tests/Test.cs (modified) (3 diffs)

branches/GBT-trunkintegration/Tests/Test.cs

@@ -76 +76 @@
           {-1.5, 20,  1},
           {-1.5, 20, -1},
-          {0.5, 20, -1},
+          {-0.5, 20, -1},
           {0.5, 10, -1},
           {1.5, 10, -1},
@@ -257 +257 @@
 
     #region helper
-    private void BuildTree(double[,] xy, string[] allVariables, int maxDepth) {
+    private void BuildTree(double[,] xy, string[] allVariables, int maxSize) {
       int nRows = xy.GetLength(0);
       var allowedInputs = allVariables.Skip(1);
@@ -266 +266 @@
       problemData.TestPartition.Start = nRows;
       problemData.TestPartition.End = nRows;
-      var rand = new MersenneTwister(31415);
-      var builder = new RegressionTreeBuilder(problemData, rand);
-      var model = (GradientBoostedTreesModel)builder.CreateRegressionTree(maxDepth, 1, 1); // maximal depth and use all rows and cols
+      var solution = GradientBoostedTreesAlgorithmStatic.TrainGbm(problemData, new SquaredErrorLoss(), maxSize, nu: 1, r: 1, m: 1, maxIterations: 1, randSeed: 31415);
+      var model = (GradientBoostedTreesModel)solution.Model;
       var treeM = model.Models.Skip(1).First() as RegressionTreeModel;