Fixes piskvorky#1828

sj29-innovate · Jan 8, 2018 · 0630101 · 0630101
1 parent f9803cb
commit 0630101
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/summarization/bm25.py b/gensim/summarization/bm25.py
@@ -122,7 +122,7 @@ def get_score(self, document, index, average_idf):
                 continue
             idf = self.idf[word] if self.idf[word] >= 0 else EPSILON * average_idf
             score += (idf * self.f[index][word] * (PARAM_K1 + 1)
-                      / (self.f[index][word] + PARAM_K1 * (1 - PARAM_B + PARAM_B * len(corpus[index]) / self.avgdl)))
+                      / (self.f[index][word] + PARAM_K1 * (1 - PARAM_B + PARAM_B * len(self.corpus[index]) / self.avgdl)))
         return score
 
     def get_scores(self, document, average_idf):