Implementing int overflow error fixes piskvorky#2700 and piskvorky#2239

tcrick · Aug 22, 2020 · 6cf7e34 · 6cf7e34
1 parent b736289
commit 6cf7e34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gensim/models/word2vec_corpusfile.pxd b/gensim/models/word2vec_corpusfile.pxd
@@ -65,5 +65,5 @@ cdef class CythonVocab:
 
 
 cdef REAL_t get_alpha(REAL_t alpha, REAL_t end_alpha, int cur_epoch, int num_epochs) nogil
-cdef REAL_t get_next_alpha(REAL_t start_alpha, REAL_t end_alpha, int total_examples, int total_words,
-                           int expected_examples, int expected_words, int cur_epoch, int num_epochs) nogil
+cdef REAL_t get_next_alpha(REAL_t start_alpha, REAL_t end_alpha, long long total_examples, long long total_words,
+                           long long expected_examples, long long expected_words, int cur_epoch, int num_epochs) nogil