Minor Changes

cooccurrence_count.py

@@ -75,12 +75,20 @@ class CooccurrenceCount(evaluate.Measurement):
         print(word1, word2)
         len1 = len(word1.split(' '))
         len2 = len(word2.split(' '))
-        gram = len1 if len1 > len2 else len2
-        print(gram)
-        v = CountVectorizer(ngram_range=(gram, gram))
+        if len1 > len2:
+            ugram = len1
+            lgram = len2
+        elif len1 < len2:
+            ugram = len2
+            lgram = len1
+        else:
+            ugram = len1
+            lgram = len1
+
+        v = CountVectorizer(ngram_range=(lgram, ugram))
         analyzer = v.build_analyzer()
         vectorizer = CountVectorizer(
-            ngram_range=(gram, gram),
+            ngram_range=(lgram, ugram),
             vocabulary={
                 analyzer(word1)[0]: 0,
                 analyzer(word2)[0]: 1