Changeset 15450 for branches/MathNetNumerics-Exploration-2789/Test/UnitTest1.cs

Timestamp:

11/06/17 13:12:41 (6 years ago)

Author:

gkronber

Message:

#2789 more tests with CV and automatic determination of smoothing parameter

File:

• branches/MathNetNumerics-Exploration-2789/Test/UnitTest1.cs (modified) (1 diff)

branches/MathNetNumerics-Exploration-2789/Test/UnitTest1.cs

@@ -110 +110 @@
 
       Console.WriteLine("Best tolerance {0}, RMSE (LOO): {1}", optTol, looRMSE);
+    }
 
+    [TestMethod]
+    public void TestGAM() {
+      var provider = new HeuristicLab.Problems.Instances.DataAnalysis.VariousInstanceProvider();
+      var problemData = provider.LoadData(provider.GetDataDescriptors().First(dd => dd.Name.Contains("Poly")));
+      // var provider = new HeuristicLab.Problems.Instances.DataAnalysis.RegressionRealWorldInstanceProvider();
+      // var problemData = provider.LoadData(provider.GetDataDescriptors().First(dd => dd.Name.Contains("Chem")));
+
+      var gam = new GAM();
+      gam.MaxIterations = 10;
+      gam.MaxInteractions = 3;
+      gam.Problem.ProblemData = problemData;
+      gam.Start();
+
+      var solution = (IRegressionSolution)gam.Results["Ensemble solution"].Value;
+
+      Console.WriteLine("RMSE (train) {0}", solution.TrainingRootMeanSquaredError);
+      Console.WriteLine("RMSE (test) {0}", solution.TestRootMeanSquaredError);
     }
-      
+
   }
 }