Changeset 15994 for branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/Analysis/RSquaredEvaluator.cs

Timestamp:

07/11/18 16:43:41 (6 years ago)

Author:

bburlacu

Message:

#2886: Add symbolic regression solution to results during algorithm run and scale model.

File:

• branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/Analysis/RSquaredEvaluator.cs (modified) (1 diff)

branches/2886_SymRegGrammarEnumeration/HeuristicLab.Algorithms.DataAnalysis.SymRegGrammarEnumeration/Analysis/RSquaredEvaluator.cs

@@ -86 +86 @@
 
       if (algorithm.BestTrainingSentence == null || r2 > bestR2 || (r2.IsAlmost(bestR2) && complexity < bestComplexity)) {
-        results.AddOrUpdateResult(BestTrainingQualityResultName, new DoubleValue(r2));
+        algorithm.BestTrainingSentence = symbolString;
+
+        var model = new SymbolicRegressionModel(problemData.TargetVariable, tree, expressionTreeLinearInterpreter);
+        model.Scale(problemData);
+        var bestSolution = model.CreateRegressionSolution(problemData);
+
+        results.AddOrUpdateResult(BestTrainingQualityResultName, new DoubleValue(bestSolution.TrainingRSquared));
+        results.AddOrUpdateResult(BestTestQualityResultName, new DoubleValue(bestSolution.TestRSquared));
+        results.AddOrUpdateResult(BestTrainingModelResultName, bestSolution.Model);
+        results.AddOrUpdateResult(BestTrainingSolutionResultName, bestSolution);
         results.AddOrUpdateResult(BestComplexityResultName, new IntValue(complexity));
-        algorithm.BestTrainingSentence = symbolString;
 
         // record best sentence quality & length