Changeset 6903 for trunk/sources

Timestamp:

10/12/11 11:27:53 (13 years ago)

Author:

epitzer

Message:

#1622 Enable discretization by empirical distribution's quantiles.

File:

• trunk/sources/HeuristicLab.Optimization/3.3/RunCollectionDiscretizer.cs (modified) (4 diffs)

trunk/sources/HeuristicLab.Optimization/3.3/RunCollectionDiscretizer.cs

@@ -50 +50 @@
     public RunCollectionDiscretizer() {
       Parameters.Add(new ValueParameter<StringValue>("Source", "Source value name to be fuzzified.", new StringValue("Value")));
-      Parameters.Add(new ValueParameter<StringValue>("Target", "Target value name. The new, fuzzified variable to be created.", new StringValue("Calc.Value")));      
-      Parameters.Add(new ValueParameter<DoubleValue>("Spread", "The number of standard deviations considered one additional level.", new DoubleValue(1)));
+      Parameters.Add(new ValueParameter<StringValue>("Target", "Target value name. The new, fuzzified variable to be created.", new StringValue("Calc.Value")));
+      Parameters.Add(new ValueParameter<DoubleValue>("Spread", "The number of standard deviations considered one additional level. Set to zero to use empirical distribution instead.", new DoubleValue(1)));
       Parameters.Add(new ValueParameter<StringValue>("GroupBy", "Create separate analyzes for different values of this variable.", new StringValue("")));
       Parameters.Add(new ValueParameter<ItemList<StringValue>>("Levels", "The list of levels to be assigned.",
@@ -101 +101 @@
         var values = group.Select(r => r.Value).ToList();
         if (values.Count > 0) {
-          var avg = values.Average();
-          var stdDev = values.StandardDeviation();
-          foreach (var r in group) {
-            r.Run.Results[Target] = new StringValue(Fuzzify(r.Value, avg, stdDev));
+          if (Spread > 0) {
+            var avg = values.Average();
+            var stdDev = values.StandardDeviation();
+            foreach (var r in group) {
+              r.Run.Results[Target] = new StringValue(Discretize(r.Value, avg, stdDev));
+            }
+          } else {
+            values.Sort();
+            var a = values.ToArray();
+            foreach (var r in group) {
+              r.Run.Results[Target] = new StringValue(Discretize(r.Value, a));
+            }
           }
         }
@@ -141 +149 @@
     }
 
-    private string Fuzzify(double value, double avg, double stdDev) {
+    private string Discretize(double value, double avg, double stdDev) {
       double dev = (value - avg)/(stdDev*Spread);
       int index;
@@ -157 +165 @@
     }
 
+    private string Discretize(double value, double[] values) {
+      var index = Array.BinarySearch(values, value);
+      var pos = 1.0*(index < 0 ? ~index : index)/(values.Length-1);
+      return Levels[Math.Min(Levels.Count - 1, Math.Max(0, (int) Math.Round(pos*(Levels.Count-1))))];
+    }
+
     #endregion
   }