diff --git a/gensim/test/test_word2vec.py b/gensim/test/test_word2vec.py index d3c4ab7295..3f3acf658e 100644 --- a/gensim/test/test_word2vec.py +++ b/gensim/test/test_word2vec.py @@ -235,9 +235,8 @@ def testPersistenceWord2VecFormatCombinationWithStandardPersistence(self): self.assertEqual(model.vocab['human'].count, binary_model_with_vocab.vocab['human'].count) # def testSaveModelsForPythonVersion(self): - # fname = os.path.join(os.path.dirname(__file__), 'word2vecmodel_python_2_7') - # model = word2vec.Word2Vec(sentences, min_count=1, seed = 1000007) - # model.init_sims() + # fname = os.path.join(os.path.dirname(__file__), 'word2vecmodel_python_3_5') + # model = word2vec.Word2Vec(sentences, size=10, min_count=0, seed=42, hs=1, negative=0) # model.save(fname) # logging.warning("Word2Vec model saved") @@ -488,11 +487,11 @@ def testRNG(self): def models_equal(self, model, model2): self.assertEqual(len(model.vocab), len(model2.vocab)) - self.assertTrue(numpy.allclose(model.syn0, model2.syn0)) + self.assertTrue(numpy.allclose(model.syn0, model2.syn0, atol=1e-4)) if model.hs: - self.assertTrue(numpy.allclose(model.syn1, model2.syn1)) + self.assertTrue(numpy.allclose(model.syn1, model2.syn1, atol=1e-4)) if model.negative: - self.assertTrue(numpy.allclose(model.syn1neg, model2.syn1neg)) + self.assertTrue(numpy.allclose(model.syn1neg, model2.syn1neg, atol=1e-4)) most_common_word = max(model.vocab.items(), key=lambda item: item[1].count)[0] self.assertTrue(numpy.allclose(model[most_common_word], model2[most_common_word])) diff --git a/gensim/test/word2vecmodel_python_2_7 b/gensim/test/word2vecmodel_python_2_7 index c95899af8e..bf73d4c63d 100644 Binary files a/gensim/test/word2vecmodel_python_2_7 and b/gensim/test/word2vecmodel_python_2_7 differ diff --git a/gensim/test/word2vecmodel_python_3_5 b/gensim/test/word2vecmodel_python_3_5 index fb4e5d6b1c..b9a30df521 100644 Binary files a/gensim/test/word2vecmodel_python_3_5 and b/gensim/test/word2vecmodel_python_3_5 differ