Changeset 13091 for branches/ClassificationModelComparison/HeuristicLab.Problems.DataAnalysis

Timestamp:

10/29/15 20:39:11 (8 years ago)

Author:

gkronber

Message:

#1998: minor changes

Location:

branches/ClassificationModelComparison/HeuristicLab.Problems.DataAnalysis/3.4/OnlineCalculators

Files:

• ConfusionMatrixCalculator.cs (modified) (2 diffs)
• FOneScoreCalculator.cs (modified) (3 diffs)
• MatthewsCorrelationCoefficientCalculator.cs (modified) (1 diff)

branches/ClassificationModelComparison/HeuristicLab.Problems.DataAnalysis/3.4/OnlineCalculators/ConfusionMatrixCalculator.cs

@@ -1 +1 @@
 #region License Information
 /* HeuristicLab
- * Copyright (C) 2002-2012 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
+ * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
  *
  * This file is part of HeuristicLab.
@@ -31 +31 @@
       }
 
-      IEnumerable<double> classValues = originalValues.Distinct();
-      IEnumerable<double> estimatedClassValues = estimatedValues.Distinct();
+      var  classValues = originalValues.Distinct().ToList();
+      var estimatedClassValues = estimatedValues.Distinct().ToList();
 
-      if (!estimatedClassValues.All(x => originalValues.Contains(x))) {
+      if (!estimatedClassValues.All(x => classValues.Contains(x))) {
         errorState = OnlineCalculatorError.InvalidValueAdded;
         return null;
       }
 
-      int classes = classValues.Count();
+      int classes = classValues.Count;
       double[,] confusionMatrix = new double[classes, classes];
 

branches/ClassificationModelComparison/HeuristicLab.Problems.DataAnalysis/3.4/OnlineCalculators/FOneScoreCalculator.cs

@@ -1 +1 @@
 #region License Information
 /* HeuristicLab
- * Copyright (C) 2002-2012 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
+ * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
  *
  * This file is part of HeuristicLab.
@@ -27 +27 @@
   public class FOneScoreCalculator {
     public static double Calculate(IEnumerable<double> originalValues, IEnumerable<double> estimatedValues, out OnlineCalculatorError errorState) {
-      if (originalValues.Distinct().Count() != 2) {
+      if (originalValues.Distinct().Skip(2).Any()) {
         throw new ArgumentException("F1 score can only be calculated for binary classification.");
       }
@@ -40 +40 @@
     private static double CalculateFOne(double[,] confusionMatrix) {
       double precision = confusionMatrix[0, 0] / (confusionMatrix[0, 0] + confusionMatrix[0, 1]);
-
       double recall = confusionMatrix[0, 0] / (confusionMatrix[0, 0] + confusionMatrix[1, 0]);
 

branches/ClassificationModelComparison/HeuristicLab.Problems.DataAnalysis/3.4/OnlineCalculators/MatthewsCorrelationCoefficientCalculator.cs

@@ -1 +1 @@
 #region License Information
 /* HeuristicLab
- * Copyright (C) 2002-2012 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
+ * Copyright (C) 2002-2015 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
  *
  * This file is part of HeuristicLab.