Ensures FastText returns consistent dtypes

piskvorky · Oct 19, 2017 · 31ef726 · 31ef726
1 parent e92b45d
commit 31ef726
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/models/wrappers/fasttext.py b/gensim/models/wrappers/fasttext.py
@@ -90,7 +90,7 @@ def word_vec(self, word, use_norm=False):
         if word in self.vocab:
             return super(FastTextKeyedVectors, self).word_vec(word, use_norm)
         else:
-            word_vec = np.zeros(self.syn0_ngrams.shape[1])
+            word_vec = np.zeros(self.syn0_ngrams.shape[1], dtype=np.float32)
             ngrams = compute_ngrams(word, self.min_n, self.max_n)
             ngrams = [ng for ng in ngrams if ng in self.ngrams]
             if use_norm: