use random_state instead of seed.

piskvorky · Jun 29, 2016 · afa0b41 · afa0b41
1 parent cc7f3eb
commit afa0b41
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/gensim/test/test_coherencemodel.py b/gensim/test/test_coherencemodel.py
@@ -45,9 +45,8 @@ def testfile():
 
 class TestCoherenceModel(unittest.TestCase):
     def setUp(self):
-        np.random.seed(8)
-        self.badLdaModel = LdaModel(corpus=corpus, num_topics=2, passes=1)  # Bad lda model
-        self.goodLdaModel = LdaModel(corpus=corpus, num_topics=2, passes=50)  # Good lda model
+        self.badLdaModel = LdaModel(corpus=corpus, num_topics=2, passes=1, random_state=17)  # Bad lda model
+        self.goodLdaModel = LdaModel(corpus=corpus, num_topics=2, passes=50, random_state=17)  # Good lda model
 
     def testUMassLdaModel(self):
         """Test U_Mass topic coherence algorithm on LDA Model"""