Changeset 13791 for branches/PerformanceComparison/HeuristicLab.OptimizationExpertSystem.Common/3.3/Recommenders/KNearestNeighborRecommender.cs

Timestamp:

04/25/16 12:56:17 (8 years ago)

Author:

abeham

Message:

#2457: working on recommendation algorithms

File:

• branches/PerformanceComparison/HeuristicLab.OptimizationExpertSystem.Common/3.3/Recommenders/KNearestNeighborRecommender.cs (modified) (3 diffs)

branches/PerformanceComparison/HeuristicLab.OptimizationExpertSystem.Common/3.3/Recommenders/KNearestNeighborRecommender.cs

@@ -23 +23 @@
 using HeuristicLab.Core;
 using HeuristicLab.Data;
+using HeuristicLab.Optimization;
 using HeuristicLab.Parameters;
 using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
+using System.Linq;
 
 namespace HeuristicLab.OptimizationExpertSystem.Common {
@@ -37 +39 @@
     [StorableConstructor]
     private KNearestNeighborRecommender(bool deserializing) : base(deserializing) { }
-    private KNearestNeighborRecommender(KNearestNeighborRecommender original, Cloner cloner)
-      : base(original, cloner) { }
+    private KNearestNeighborRecommender(KNearestNeighborRecommender original, Cloner cloner) : base(original, cloner) { }
     public KNearestNeighborRecommender() {
-      Parameters.Add(new FixedValueParameter<IntValue>("K", "The number of nearest neighbors to consider.", new IntValue(5)));
+      Parameters.Add(new FixedValueParameter<IntValue>("K", "The number of nearest neighbors to consider.", new IntValue(3)));
     }
 
@@ -47 +48 @@
     }
 
-    public IRecommendationModel TrainModel(KnowledgeCenter kc, string[] characteristics) {
-      return new KNearestNeighborModel(KParameter.Value.Value, characteristics);
+    public IRecommendationModel TrainModel(IRun[] problemInstances, KnowledgeCenter kc, string[] characteristics) {
+      var perfData = problemInstances.Select(pi => new { ProblemInstance = pi, Performance = kc.GetAlgorithmPerformance(pi) })
+                                     .ToDictionary(x => x.ProblemInstance, x => x.Performance);
+      return new KNearestNeighborModel(KParameter.Value.Value, perfData, characteristics);
     }
   }