Changeset 14474 for branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing/3.4/Content/ScatterPlotContent.cs

Timestamp:

12/12/16 11:17:07 (7 years ago)

Author:

pfleck

Message:

#2709

• Improved legend description for grouped histogram and scatterplots.
• Fixed initial size of points for scatterplots.
• Added correlation calculation for scatterplots (not used yet).

File:

• branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing/3.4/Content/ScatterPlotContent.cs (modified) (5 diffs)

branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing/3.4/Content/ScatterPlotContent.cs

@@ -25 +25 @@
 using HeuristicLab.Analysis;
 using HeuristicLab.Common;
+using HeuristicLab.Problems.DataAnalysis;
 using HeuristicLab.Visualization.ChartControlsExtensions;
 
@@ -38 +39 @@
     }
 
-    public ScatterPlot CreateScatterPlot(string variableNameX, string variableNameY, string variableNameColor = "-") {
+    public ScatterPlot CreateScatterPlot(string variableNameX, string variableNameY, string variableNameGroup = "-") {
       ScatterPlot scatterPlot = new ScatterPlot();
 
@@ -60 +61 @@
       } catch (ArgumentOutOfRangeException) { } // missing values lead to NaNs
 
-      if (variableNameColor == null || variableNameColor == "-") {
+      if (variableNameGroup == null || variableNameGroup == "-") {
         List<Point2D<double>> points = new List<Point2D<double>>();
 
@@ -68 +69 @@
         }
 
-        ScatterPlotDataRow scdr = new ScatterPlotDataRow(variableNameX + " - " + variableNameY, "", points);
+        ScatterPlotDataRow scdr = new ScatterPlotDataRow(variableNameX + " - " + variableNameY, "", points) {
+          VisualProperties = { PointSize = 6 }
+        };
         scdr.VisualProperties.IsVisibleInLegend = false;
         scatterPlot.Rows.Add(scdr);
 
       } else {
-        var colorValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameColor));
-        var data = xValues.Zip(yValues, (x, y) => new { x, y }).Zip(colorValues, (v, c) => new { v.x, v.y, c }).ToList();
-        var gradients = ColorGradient.Colors;
-        int curGradient = 0;
-        int numColors = colorValues.Distinct().Count();
-        foreach (var colorValue in colorValues.Distinct()) {
-          var values = data.Where(x => x.c == colorValue);
+        var groupValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameGroup));
+        var data = xValues.Zip(yValues, (x, y) => new { x, y }).Zip(groupValues, (v, c) => new { v.x, v.y, c }).ToList();
+        foreach (var groupValue in groupValues.Distinct()) {
+          var values = data.Where(x => x.c == groupValue);
           var row = new ScatterPlotDataRow(
-            variableNameX + " - " + variableNameY + " (" + colorValue + ")",
+            variableNameGroup + " (" + groupValue + ")",
             "",
-            values.Select(v => new Point2D<double>(v.x, v.y)),
-            new ScatterPlotDataRowVisualProperties() { Color = gradients[curGradient] });
-          curGradient += gradients.Count / numColors;
+            values.Select(v => new Point2D<double>(v.x, v.y))) {
+            VisualProperties = { PointSize = 6 }
+          };
           scatterPlot.Rows.Add(row);
         }
@@ -91 +91 @@
       return scatterPlot;
     }
+
+    public DataRow GetCorrelationRow(string variableNameX, string variableNameY) {
+      var xValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameX));
+      var yValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameY));
+
+      double k, d;
+      OnlineCalculatorError err;
+      OnlineLinearScalingParameterCalculator.Calculate(xValues, yValues, out k, out d, out err);
+      double p = OnlinePearsonsRCalculator.Calculate(xValues, yValues, out err);
+
+      var data = new double[xValues.Count];
+      for (int i = 0; i < xValues.Count; i++) {
+        data[i]= k * i + d;
+      }
+
+      return new DataRow(string.Format("Correlation (R²={0})", p*p), "", data);
+    }
   }
 }