Changeset 8534 for branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation

Timestamp:

08/28/12 16:13:02 (12 years ago)

Author:

sforsten

Message:

#1776:

• merged r8508:8533 from trunk into branch
• AverageThresholdCalculator and MedianThresholdCalculator can now handle multi class classification
• changed combo boxes in ClassificationEnsembleSolutionView to drop down list

Location:

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis

Files:

• . (modified) (1 prop)
• 3.4/Implementation/Classification/ClassificationEnsembleModel.cs (modified) (1 diff)
• 3.4/Implementation/Classification/ClassificationEnsembleSolution.cs (modified) (1 diff)
• 3.4/Implementation/Classification/ClassificationProblemData.cs (modified) (1 diff)
• 3.4/Implementation/Classification/DiscriminantFunctionClassificationSolution.cs (modified) (1 diff)
• 3.4/Implementation/Classification/DiscriminantFunctionClassificationSolutionBase.cs (modified) (2 diffs)
• 3.4/Implementation/Classification/WeightCalculators/AverageThresholdCalculator.cs (modified) (3 diffs)
• 3.4/Implementation/Classification/WeightCalculators/MedianThresholdCalculator.cs (modified) (3 diffs)
• 3.4/Implementation/Regression/ConstantRegressionModel.cs (modified) (1 diff)
• 3.4/Implementation/Regression/RegressionEnsembleModel.cs (modified) (1 diff)
• 3.4/Implementation/Regression/RegressionProblemData.cs (modified) (1 diff)

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/ClassificationEnsembleModel.cs

@@ -95 +95 @@
 
     IClassificationSolution IClassificationModel.CreateClassificationSolution(IClassificationProblemData problemData) {
-      return new ClassificationEnsembleSolution(models, problemData);
+      return new ClassificationEnsembleSolution(models, new ClassificationEnsembleProblemData(problemData));
     }
     #endregion

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/ClassificationEnsembleSolution.cs

@@ -114 +114 @@
     }
 
+    public ClassificationEnsembleSolution(IClassificationProblemData problemData) :
+      this(Enumerable.Empty<IClassificationModel>(), problemData) { }
+
     public ClassificationEnsembleSolution(IEnumerable<IClassificationModel> models, IClassificationProblemData problemData)
       : this(models, problemData,

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/ClassificationProblemData.cs

@@ -277 +277 @@
 
     public ClassificationProblemData() : this(defaultDataset, defaultAllowedInputVariables, defaultTargetVariable) { }
+
+    public ClassificationProblemData(IClassificationProblemData classificationProblemData)
+      : this(classificationProblemData.Dataset, classificationProblemData.AllowedInputVariables, classificationProblemData.TargetVariable) {
+      TrainingPartition.Start = classificationProblemData.TrainingPartition.Start;
+      TrainingPartition.End = classificationProblemData.TrainingPartition.End;
+      TestPartition.Start = classificationProblemData.TestPartition.Start;
+      TestPartition.End = classificationProblemData.TestPartition.End;
+    }
+
     public ClassificationProblemData(Dataset dataset, IEnumerable<string> allowedInputVariables, string targetVariable)
       : base(dataset, allowedInputVariables) {

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/DiscriminantFunctionClassificationSolution.cs

@@ -51 +51 @@
       valueEvaluationCache = new Dictionary<int, double>();
       classValueEvaluationCache = new Dictionary<int, double>();
-
-      SetAccuracyMaximizingThresholds();
     }
 

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/DiscriminantFunctionClassificationSolutionBase.cs

@@ -96 +96 @@
     protected override void OnModelChanged() {
       DeregisterEventHandler();
-      SetAccuracyMaximizingThresholds();
       RegisterEventHandler();
       base.OnModelChanged();
@@ -137 +136 @@
     }
 
-    public void SetAccuracyMaximizingThresholds() {
-      double[] classValues;
-      double[] thresholds;
-      var targetClassValues = ProblemData.Dataset.GetDoubleValues(ProblemData.TargetVariable, ProblemData.TrainingIndices);
-      AccuracyMaximizationThresholdCalculator.CalculateThresholds(ProblemData, EstimatedTrainingValues, targetClassValues, out classValues, out thresholds);
-
-      Model.SetThresholdsAndClassValues(thresholds, classValues);
-    }
-
-    public void SetClassDistibutionCutPointThresholds() {
-      double[] classValues;
-      double[] thresholds;
-      var targetClassValues = ProblemData.Dataset.GetDoubleValues(ProblemData.TargetVariable, ProblemData.TrainingIndices);
-      NormalDistributionCutPointsThresholdCalculator.CalculateThresholds(ProblemData, EstimatedTrainingValues, targetClassValues, out classValues, out thresholds);
-
-      Model.SetThresholdsAndClassValues(thresholds, classValues);
-    }
-
     protected virtual void OnModelThresholdsChanged(EventArgs e) {
       CalculateResults();

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/WeightCalculators/AverageThresholdCalculator.cs

@@ -58 +58 @@
 
     protected override double GetDiscriminantConfidence(IEnumerable<IDiscriminantFunctionClassificationSolution> solutions, int index, double estimatedClassValue) {
-      // only works with binary classification
-      if (!classValues.Count().Equals(2))
-        return double.NaN;
       Dataset dataset = solutions.First().ProblemData.Dataset;
       IList<double> values = solutions.Select(s => s.Model.GetEstimatedValues(dataset, Enumerable.Repeat(index, 1)).First()).ToList();
@@ -70 +67 @@
 
     public override IEnumerable<double> GetDiscriminantConfidence(IEnumerable<IDiscriminantFunctionClassificationSolution> solutions, IEnumerable<int> indices, IEnumerable<double> estimatedClassValue) {
-      if (!classValues.Count().Equals(2))
-        return Enumerable.Repeat(double.NaN, indices.Count());
-
       Dataset dataset = solutions.First().ProblemData.Dataset;
       double[][] values = solutions.Select(s => s.Model.GetEstimatedValues(dataset, indices).ToArray()).ToArray();
@@ -87 +81 @@
 
     protected double GetAverageConfidence(double avg, double estimatedClassValue) {
-      if (estimatedClassValue.Equals(classValues[0])) {
-        if (avg < estimatedClassValue)
-          return 1;
-        else if (avg >= threshold[1])
-          return 0;
-        else {
-          double distance = threshold[1] - classValues[0];
-          return (1 / distance) * (threshold[1] - avg);
+      for (int i = 0; i < classValues.Length; i++) {
+        if (estimatedClassValue.Equals(classValues[i])) {
+          //special case: avgerage is higher than value of highest class
+          if (i == classValues.Length - 1 && avg > estimatedClassValue) {
+            return 1;
+          }
+          //special case: average is lower than value of lowest class
+          if (i == 0 && avg < estimatedClassValue) {
+            return 1;
+          }
+          //special case: average is not between threshold of estimated class value
+          if ((i < classValues.Length - 1 && avg >= threshold[i + 1]) || avg <= threshold[i]) {
+            return 0;
+          }
+
+          double thresholdToClassDistance, thresholdToAverageValueDistance;
+          if (avg >= classValues[i]) {
+            thresholdToClassDistance = threshold[i + 1] - classValues[i];
+            thresholdToAverageValueDistance = threshold[i + 1] - avg;
+          } else {
+            thresholdToClassDistance = classValues[i] - threshold[i];
+            thresholdToAverageValueDistance = avg - threshold[i];
+          }
+          return (1 / thresholdToClassDistance) * thresholdToAverageValueDistance;
         }
-      } else if (estimatedClassValue.Equals(classValues[1])) {
-        if (avg > estimatedClassValue)
-          return 1;
-        else if (avg <= threshold[1])
-          return 0;
-        else {
-          double distance = classValues[1] - threshold[1];
-          return (1 / distance) * (avg - threshold[1]);
-        }
-      } else
-        return double.NaN;
+      }
+      return double.NaN;
     }
   }

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Classification/WeightCalculators/MedianThresholdCalculator.cs

@@ -58 +58 @@
 
     protected override double GetDiscriminantConfidence(IEnumerable<IDiscriminantFunctionClassificationSolution> solutions, int index, double estimatedClassValue) {
-      // only works with binary classification
-      if (!classValues.Count().Equals(2))
-        return double.NaN;
+
       Dataset dataset = solutions.First().ProblemData.Dataset;
       IList<double> values = solutions.Select(s => s.Model.GetEstimatedValues(dataset, Enumerable.Repeat(index, 1)).First()).ToList();
@@ -70 +68 @@
 
     public override IEnumerable<double> GetDiscriminantConfidence(IEnumerable<IDiscriminantFunctionClassificationSolution> solutions, IEnumerable<int> indices, IEnumerable<double> estimatedClassValue) {
-      if (!classValues.Count().Equals(2))
-        return Enumerable.Repeat(double.NaN, indices.Count());
 
       Dataset dataset = solutions.First().ProblemData.Dataset;
@@ -87 +83 @@
 
     protected double GetMedianConfidence(double median, double estimatedClassValue) {
-      if (estimatedClassValue.Equals(classValues[0])) {
-        if (median < estimatedClassValue)
-          return 1;
-        else if (median >= threshold[1])
-          return 0;
-        else {
-          double distance = threshold[1] - classValues[0];
-          return (1 / distance) * (threshold[1] - median);
+      for (int i = 0; i < classValues.Length; i++) {
+        if (estimatedClassValue.Equals(classValues[i])) {
+          //special case: avgerage is higher than value of highest class
+          if (i == classValues.Length - 1 && median > estimatedClassValue) {
+            return 1;
+          }
+          //special case: average is lower than value of lowest class
+          if (i == 0 && median < estimatedClassValue) {
+            return 1;
+          }
+          //special case: average is not between threshold of estimated class value
+          if ((i < classValues.Length - 1 && median >= threshold[i + 1]) || median <= threshold[i]) {
+            return 0;
+          }
+
+          double thresholdToClassDistance, thresholdToAverageValueDistance;
+          if (median >= classValues[i]) {
+            thresholdToClassDistance = threshold[i + 1] - classValues[i];
+            thresholdToAverageValueDistance = threshold[i + 1] - median;
+          } else {
+            thresholdToClassDistance = classValues[i] - threshold[i];
+            thresholdToAverageValueDistance = median - threshold[i];
+          }
+          return (1 / thresholdToClassDistance) * thresholdToAverageValueDistance;
         }
-      } else if (estimatedClassValue.Equals(classValues[1])) {
-        if (median > estimatedClassValue)
-          return 1;
-        else if (median <= threshold[1])
-          return 0;
-        else {
-          double distance = classValues[1] - threshold[1];
-          return (1 / distance) * (median - threshold[1]);
-        }
-      } else
-        return double.NaN;
+      }
+      return double.NaN;
     }
 

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Regression/ConstantRegressionModel.cs

@@ -55 +55 @@
 
     public IRegressionSolution CreateRegressionSolution(IRegressionProblemData problemData) {
-      return new ConstantRegressionSolution(this, problemData);
+      return new ConstantRegressionSolution(this, new RegressionProblemData(problemData));
     }
   }

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Regression/RegressionEnsembleModel.cs

@@ -102 +102 @@
 
     public RegressionEnsembleSolution CreateRegressionSolution(IRegressionProblemData problemData) {
-      return new RegressionEnsembleSolution(this.Models, problemData);
+      return new RegressionEnsembleSolution(this.Models, new RegressionEnsembleProblemData(problemData));
     }
     IRegressionSolution IRegressionModel.CreateRegressionSolution(IRegressionProblemData problemData) {

branches/ClassificationEnsembleVoting/HeuristicLab.Problems.DataAnalysis/3.4/Implementation/Regression/RegressionProblemData.cs

@@ -121 +121 @@
       : this(defaultDataset, defaultAllowedInputVariables, defaultTargetVariable) {
     }
+    public RegressionProblemData(IRegressionProblemData regressionProblemData)
+      : this(regressionProblemData.Dataset, regressionProblemData.AllowedInputVariables, regressionProblemData.TargetVariable) {
+      TrainingPartition.Start = regressionProblemData.TrainingPartition.Start;
+      TrainingPartition.End = regressionProblemData.TrainingPartition.End;
+      TestPartition.Start = regressionProblemData.TestPartition.Start;
+      TestPartition.End = regressionProblemData.TestPartition.End;
+    }
 
     public RegressionProblemData(Dataset dataset, IEnumerable<string> allowedInputVariables, string targetVariable)