Minor Changes

cooccurrence_count.py

@@ -86,13 +86,11 @@ class CooccurrenceCount(evaluate.Measurement):
 
         v = CountVectorizer(ngram_range=(lgram, ugram))
         analyzer = v.build_analyzer()
-        print(analyzer(word1))
-        print(analyzer(word2))
         vectorizer = CountVectorizer(
             ngram_range=(lgram, ugram),
             vocabulary={
-                analyzer(word1)[0]: 0,
-                analyzer(word2)[0]: 1
+                analyzer(word1)[-1]: 0,
+                analyzer(word2)[-1]: 1
             }
         )
         co_occurrences = vectorizer.fit_transform(data)