From f6adac5c30b582ec3ccdfaa488c5996baabf8582 Mon Sep 17 00:00:00 2001 From: anmol01gulati Date: Fri, 11 Nov 2016 09:52:09 +0530 Subject: [PATCH] Increased window size in test_cbow_hs_online --- gensim/test/test_word2vec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gensim/test/test_word2vec.py b/gensim/test/test_word2vec.py index dc8e221294..8a59ca2c03 100644 --- a/gensim/test/test_word2vec.py +++ b/gensim/test/test_word2vec.py @@ -124,7 +124,7 @@ def test_sg_neg_online(self): def test_cbow_hs_online(self): """Test CBOW w/ hierarchical softmax""" - model = word2vec.Word2Vec(sg=0, cbow_mean=1, alpha=0.05, window=3, hs=1, negative=0, + model = word2vec.Word2Vec(sg=0, cbow_mean=1, alpha=0.05, window=5, hs=1, negative=0, min_count=3, iter=10, seed=42, workers=2) self.onlineSanity(model)