Changeset 16416 for branches/2904_CalculateImpacts/HeuristicLab.Tests/HeuristicLab.Problems.DataAnalysis-3.4/ClassificationVariableImpactCalculationTest.cs

Timestamp:

12/20/18 11:11:37 (6 years ago)

Author:

fholzing

Message:

#2904: cleared variableImpacts on OnContentChanged(); Used existing epsilon (almost) function for comparison of estimated impacts; removed direct usage of alglib; Added a LDA- and RF-Unittest for Impact-Calculation; Changed the approach of loading ProblemData (no more strings); Renamed PerformanceTests for Regression/Classification so it can be found easier

File:

• branches/2904_CalculateImpacts/HeuristicLab.Tests/HeuristicLab.Problems.DataAnalysis-3.4/ClassificationVariableImpactCalculationTest.cs (modified) (6 diffs)

branches/2904_CalculateImpacts/HeuristicLab.Tests/HeuristicLab.Problems.DataAnalysis-3.4/ClassificationVariableImpactCalculationTest.cs

@@ -26 +26 @@
     }
 
-    private static readonly double epsilon = 0.00001;
 
     [TestMethod]
@@ -48 +47 @@
       ClassificationSolutionVariableImpactsCalculator.CalculateImpacts(solution);
       Dictionary<string, double> expectedImpacts = GetExpectedValuesForIrisKNNModel();
+
+      CheckDefaultAsserts(solution, expectedImpacts);
+    }
+
+
+    [TestMethod]
+    [TestCategory("Problems.DataAnalysis")]
+    [TestProperty("Time", "short")]
+    public void LDAIrisVariableImpactTest() {
+      IClassificationProblemData problemData = LoadIrisProblem();
+      IClassificationSolution solution = LinearDiscriminantAnalysis.CreateLinearDiscriminantAnalysisSolution(problemData);
+      ClassificationSolutionVariableImpactsCalculator.CalculateImpacts(solution);
+      Dictionary<string, double> expectedImpacts = GetExpectedValuesForIrisLDAModel();
 
       CheckDefaultAsserts(solution, expectedImpacts);
@@ -85 +97 @@
     [TestProperty("Time", "short")]
     [ExpectedException(typeof(ArgumentException))]
-    public void WrongDataSetTest() {
+    public void WrongDataSetVariableImpactClassificationTest() {
       IClassificationProblemData problemData = LoadIrisProblem();
       IClassificationSolution solution = NearestNeighbourClassification.CreateNearestNeighbourClassificationSolution(problemData, 3);
@@ -99 +111 @@
     [TestCategory("Problems.DataAnalysis")]
     [TestProperty("Time", "medium")]
-    public void PerformanceTest() {
+    public void PerformanceVariableImpactClassificationTest() {
       int rows = 1500;
       int columns = 77;
@@ -229 +241 @@
       return expectedImpacts;
     }
+    private Dictionary<string, double> GetExpectedValuesForIrisLDAModel() {
+      Dictionary<string, double> expectedImpacts = new Dictionary<string, double>();
+      expectedImpacts.Add("sepal_width", 0.01);
+      expectedImpacts.Add("sepal_length", 0.03);
+      expectedImpacts.Add("petal_width", 0.2);
+      expectedImpacts.Add("petal_length", 0.5);
+
+      return expectedImpacts;
+    }
     #endregion
 
@@ -243 +264 @@
       //Check if impacts are as expected
       Assert.AreEqual(modelImpacts.Count(), expectedImpacts.Count);
-      Assert.IsTrue(modelImpacts.All(v => Math.Abs(expectedImpacts[v.Item1] - v.Item2) < epsilon));
+      Assert.IsTrue(modelImpacts.All(v => v.Item2.IsAlmost(expectedImpacts[v.Item1])));
     }
   }