Prabin Bhandari commited on
Commit
2c03ef4
1 Parent(s): 8b1de7e

Minor Changes

Browse files
Files changed (1) hide show
  1. cooccurrence_count.py +2 -4
cooccurrence_count.py CHANGED
@@ -86,13 +86,11 @@ class CooccurrenceCount(evaluate.Measurement):
86
 
87
  v = CountVectorizer(ngram_range=(lgram, ugram))
88
  analyzer = v.build_analyzer()
89
- print(analyzer(word1))
90
- print(analyzer(word2))
91
  vectorizer = CountVectorizer(
92
  ngram_range=(lgram, ugram),
93
  vocabulary={
94
- analyzer(word1)[0]: 0,
95
- analyzer(word2)[0]: 1
96
  }
97
  )
98
  co_occurrences = vectorizer.fit_transform(data)
 
86
 
87
  v = CountVectorizer(ngram_range=(lgram, ugram))
88
  analyzer = v.build_analyzer()
 
 
89
  vectorizer = CountVectorizer(
90
  ngram_range=(lgram, ugram),
91
  vocabulary={
92
+ analyzer(word1)[-1]: 0,
93
+ analyzer(word2)[-1]: 1
94
  }
95
  )
96
  co_occurrences = vectorizer.fit_transform(data)