X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=norms.py;h=eb436c3b8163141a3ada1f1f02f8be741d6f47fb;hb=533fd44ecc7048fa6480b1f6f4488088b3399abc;hp=b8e4bf1ef82c8b8ea6f85c0e3c3597d90b7d9b8b;hpb=96d46a680a808555a9ff77f2eaa68383569f07ee;p=cipher-training.git

diff --git a/norms.py b/norms.py
index b8e4bf1..eb436c3 100644
--- a/norms.py
+++ b/norms.py
@@ -13,7 +13,7 @@ def normalise(frequencies):
     >>> sorted(normalise({1: 1, 2: 2, 3: 1}).items())
     [(1, 0.25), (2, 0.5), (3, 0.25)]
     """
-    length = sum([f for f in frequencies.values()])
+    length = sum(f for f in frequencies.values())
     return collections.defaultdict(int, ((k, v / length) 
         for (k, v) in frequencies.items()))
 
@@ -178,7 +178,7 @@ def cosine_similarity(frequencies1, frequencies2):
     for k in frequencies1:
         numerator += frequencies1[k] * frequencies2[k]
         length1 += frequencies1[k]**2
-    for k in frequencies2.keys():
+    for k in frequencies2:
         length2 += frequencies2[k]**2
     return numerator / (length1 ** 0.5 * length2 ** 0.5)