Tweak: use generator comprehension.

darius · darius · commit e6c9b291a110 · 2011-10-23T21:00:01.000Z
diff --git a/learning.py b/learning.py
@@ -206,7 +206,7 @@ def predict(example):
         "Find the k closest, and have them vote for the best."
         best = heapq.nsmallest(k, ((dataset.distance(e, example), e)
                                    for e in dataset.examples))
-        return mode([e[dataset.target] for (d, e) in best])
+        return mode(e[dataset.target] for (d, e) in best)
     return predict
 
 #______________________________________________________________________________
