Merge pull request #34 from several27/patch-2

Fixes wrong var assignment
IntelLabs · Feb 27, 2019 · 6d64b63 · 6d64b63
2 parents 6bc7c0e + 9ef1c26
commit 6d64b63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nlp_architect/models/np2vec.py b/nlp_architect/models/np2vec.py
@@ -211,7 +211,7 @@ def _train(self):
                 hs=self.hs,
                 negative=self.negative,
                 cbow_mean=self.cbow_mean,
-                iter=iter,
+                iter=self.iter,
                 min_n=self.min_n,
                 max_n=self.max_n,
                 word_ngrams=self.word_ngrams)