Changeset 14104 for branches/HeuristicLab.OSGAEvaluator/HeuristicLab.OSGAEvaluator/OSGAPredictionCountsAnalyzer.cs

Timestamp:

07/18/16 17:14:33 (8 years ago)

Author:

bburlacu

Message:

#2635: Improve performance and accuracy of evaluator and analyzer.

File:

• branches/HeuristicLab.OSGAEvaluator/HeuristicLab.OSGAEvaluator/OSGAPredictionCountsAnalyzer.cs (modified) (3 diffs)

branches/HeuristicLab.OSGAEvaluator/HeuristicLab.OSGAEvaluator/OSGAPredictionCountsAnalyzer.cs

@@ -1 @@
-using HeuristicLab.Common;
+using System.Linq;
+using HeuristicLab.Analysis;
+using HeuristicLab.Common;
 using HeuristicLab.Core;
+using HeuristicLab.Optimization;
+using HeuristicLab.Parameters;
 using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
 using HeuristicLab.Problems.DataAnalysis.Symbolic;
+using HeuristicLab.Problems.DataAnalysis.Symbolic.Regression;
 
 namespace HeuristicLab.OSGAEvaluator {
@@ -9 +14 @@
 
   public class OSGAPredictionCountsAnalyzer : SymbolicDataAnalysisSingleObjectiveAnalyzer {
+    private const string EvaluatorParameterName = "Evaluator";
+    private const string ResultCollectionParameterName = "Results";
+
+    public ILookupParameter<SymbolicRegressionSingleObjectiveOsgaEvaluator> EvaluatorParameter {
+      get { return (ILookupParameter<SymbolicRegressionSingleObjectiveOsgaEvaluator>)Parameters[EvaluatorParameterName]; }
+    }
+
+    public OSGAPredictionCountsAnalyzer() {
+      Parameters.Add(new LookupParameter<SymbolicRegressionSingleObjectiveOsgaEvaluator>(EvaluatorParameterName));
+    }
 
     protected OSGAPredictionCountsAnalyzer(OSGAPredictionCountsAnalyzer original, Cloner cloner) : base(original, cloner) {
@@ -16 +31 @@
       return new OSGAPredictionCountsAnalyzer(this, cloner);
     }
+
+    public override IOperation Apply() {
+      var evaluator = EvaluatorParameter.ActualValue;
+      if (evaluator == null)
+        return base.Apply();
+
+      var rejectedStats = evaluator.RejectedStats;
+      var totalStats = evaluator.TotalStats;
+
+      if (rejectedStats.Rows == 0 || totalStats.Rows == 0)
+        return base.Apply();
+
+      ResultCollection predictionResults;
+      DataTable rejectedStatsTable, totalStatsTable;
+      if (!ResultCollection.ContainsKey("OS Prediction")) {
+        predictionResults = new ResultCollection();
+        rejectedStatsTable = new DataTable("Rejected Stats");
+        foreach (var rowName in rejectedStats.RowNames) {
+          rejectedStatsTable.Rows.Add(new DataRow(rowName) { VisualProperties = { StartIndexZero = true, ChartType = DataRowVisualProperties.DataRowChartType.Columns, LineWidth = 1 } });
+        }
+        predictionResults.Add(new Result("Rejected Stats", rejectedStatsTable));
+        totalStatsTable = new DataTable("Total Stats");
+        foreach (var rowName in totalStats.RowNames) {
+          totalStatsTable.Rows.Add(new DataRow(rowName) { VisualProperties = { StartIndexZero = true, ChartType = DataRowVisualProperties.DataRowChartType.Columns, LineWidth = 1 } });
+        }
+        predictionResults.Add(new Result("Total Stats", totalStatsTable));
+
+        ResultCollection.Add(new Result("OS Prediction", predictionResults));
+      } else {
+        predictionResults = (ResultCollection)ResultCollection["OS Prediction"].Value;
+        rejectedStatsTable = (DataTable)predictionResults["Rejected Stats"].Value;
+        totalStatsTable = (DataTable)predictionResults["Total Stats"].Value;
+      }
+
+      int i = 0;
+      foreach (var rowName in rejectedStats.RowNames) {
+        // add a padding zero below to prevent columns from being cut off to the left
+        rejectedStatsTable.Rows[rowName].Values.Replace(new[] { 0d }.Concat(Enumerable.Range(0, rejectedStats.Columns).Select(j => (double)rejectedStats[i, j])).Concat(new[] { 0d }));
+        ++i;
+      }
+      i = 0;
+      foreach (var rowName in totalStats.RowNames) {
+        totalStatsTable.Rows[rowName].Values.Replace(new[] { 0d }.Concat(Enumerable.Range(0, totalStats.Columns).Select(j => (double)totalStats[i, j])).Concat(new[] { 0d }));
+        ++i;
+      }
+      return base.Apply();
+    }
   }
 }