Changeset 13797 for branches/PerformanceComparison/HeuristicLab.Analysis/3.3/Clustering/ClusterHelper.cs

Timestamp:

04/26/16 13:07:26 (8 years ago)

Author:

abeham

Message:

#2457: worked on testing recommendation algorithms through x-validation

File:

• branches/PerformanceComparison/HeuristicLab.Analysis/3.3/Clustering/ClusterHelper.cs (modified) (3 diffs)

branches/PerformanceComparison/HeuristicLab.Analysis/3.3/Clustering/ClusterHelper.cs

@@ -31 +31 @@
     private List<KeyValuePair<T, double>> excluded;
     private int[] clusterValues;
+    private int[] rankedMap;
 
     private ClusteringHelper(int K) {
@@ -50 +51 @@
       var helper = new ClusteringHelper<T>(k);
       helper.Initialize(values, excludeFunc);
-      if (helper.instances.Count == 0) 
+      if (helper.instances.Count == 0) {
         helper.clusterValues = new int[0];
-      else CkMeans1D.Cluster(helper.instances.Select(x => x.Value).ToArray(), k, out helper.clusterValues);
+        helper.rankedMap = new int[k + 1];
+      } else {
+        var centroids = CkMeans1D.Cluster(helper.instances.Select(x => x.Value).ToArray(), k, out helper.clusterValues);
+        helper.rankedMap = new int[k + 1];
+        var rank = 0;
+        foreach (var c in centroids) helper.rankedMap[c.Value] = rank++;
+      }
+      // excluded are always ranked last
+      helper.rankedMap[k] = k;
       return helper;
     }
@@ -61 +70 @@
     /// <returns></returns>
     public IEnumerable<KeyValuePair<T, Tuple<double, int>>> GetByInstance() {
-      return GetClustered().Select(x => new KeyValuePair<T, Tuple<double, int>>(x.Item1.Key, Tuple.Create(x.Item1.Value, x.Item2)));
+      return GetClustered().Select(x => new KeyValuePair<T, Tuple<double, int>>(x.Item1.Key, Tuple.Create(x.Item1.Value, rankedMap[x.Item2])));
     }
 
     public IEnumerable<KeyValuePair<int, List<KeyValuePair<T, double>>>> GetByCluster() {
       return GetClustered().GroupBy(x => x.Item2)
-        .Select(x => new KeyValuePair<int, List<KeyValuePair<T, double>>>(x.Key, x.Select(y => y.Item1).ToList()));
+        .Select(x => new KeyValuePair<int, List<KeyValuePair<T, double>>>(rankedMap[x.Key], x.Select(y => y.Item1).ToList()));
     }