removing metric organization from experiment

Aznoryusof · Mar 21, 2019 · 90cb8cf · 90cb8cf
1 parent 9f95a0b
commit 90cb8cf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cornac/experiment/experiment.py b/cornac/experiment/experiment.py
@@ -91,16 +91,16 @@ def run(self):
         organized_metrics = {'ranking': [], 'rating': []}
 
         # Organize metrics into "rating" and "ranking" for efficiency purposes
-        for mt in self.metrics:
-            organized_metrics[mt.type].append(mt)
-            metric_names.append(mt.name)
+        #for mt in self.metrics:
+        #    organized_metrics[mt.type].append(mt)
+        #    metric_names.append(mt.name)
 
         for model in self.models:
             if self.verbose:
                 print(model.name)
 
             model_names.append(model.name)
-            model_res = self.eval_method.evaluate(model=model, metrics=organized_metrics, user_based=self.user_based)
+            model_res = self.eval_method.evaluate(model=model, metrics=self.metrics, user_based=self.user_based)
             model_res._organize_avg_res(model_name=model.name, metric_names=metric_names)
             self.results._add_model_res(res=model_res, model_name=model.name)