Changeset 332 for trunk/sources/HeuristicLab.StructureIdentification/Evaluation/MeanSquaredErrorEvaluator.cs

Timestamp:

06/23/08 19:13:17 (16 years ago)

Author:

gkronber

Message:

implemented #172

File:

• trunk/sources/HeuristicLab.StructureIdentification/Evaluation/MeanSquaredErrorEvaluator.cs (modified) (5 diffs)

trunk/sources/HeuristicLab.StructureIdentification/Evaluation/MeanSquaredErrorEvaluator.cs

@@ -32 +32 @@
 namespace HeuristicLab.StructureIdentification {
   public class MeanSquaredErrorEvaluator : GPEvaluatorBase {
+    protected double[] backupValues;
     public override string Description {
       get {
@@ -41 +42 @@
     public MeanSquaredErrorEvaluator()
       : base() {
+      AddVariableInfo(new VariableInfo("UseEstimatedTargetValue", "Wether to use the original (measured) or the estimated (calculated) value for the targat variable when doing autoregressive modelling", typeof(BoolData), VariableKind.In));
+      GetVariableInfo("UseEstimatedTargetValue").Local = true;
+      AddVariable(new HeuristicLab.Core.Variable("UseEstimatedTargetValue", new BoolData(false)));
     }
 
@@ -46 +50 @@
       double errorsSquaredSum = 0;
       double targetMean = dataset.GetMean(targetVariable);
+      bool useEstimatedValues = GetVariableValue<BoolData>("UseEstimatedTargetValue", scope, false).Data;
+      if(useEstimatedValues && backupValues == null) {
+        backupValues = new double[dataset.Rows];
+        for(int i = 0; i < dataset.Rows; i++) {
+          backupValues[i] = dataset.GetValue(i, targetVariable);
+        }
+      }
+
       for(int sample = 0; sample < dataset.Rows; sample++) {
-
         double estimated = functionTree.Evaluate(dataset, sample);
         double original = dataset.GetValue(sample, targetVariable);
@@ -59 +70 @@
         double error = estimated - original;
         errorsSquaredSum += error * error;
+        if(useEstimatedValues) {
+          dataset.SetValue(sample, targetVariable, estimated);
+        }
       }
+
+      if(useEstimatedValues) RestoreDataset(dataset, targetVariable);
       errorsSquaredSum /= dataset.Rows;
       if(double.IsNaN(errorsSquaredSum) || double.IsInfinity(errorsSquaredSum)) {
@@ -67 +83 @@
       return errorsSquaredSum;
     }
+
+    private void RestoreDataset(Dataset dataset, int targetVariable) {
+      for(int i = 0; i < dataset.Rows; i++) {
+        dataset.SetValue(i, targetVariable, backupValues[i]);
+      }
+    }
   }
 }