Explicit epochs and corpus size in word2vec train(). Continuing #1139. …

…Fix #1052. (#1237) * fix the compatibility between python2 & 3 * require explicit corpus size, epochs for train() * make all train() calls use explicit count, epochs * add tests to make sure that ValueError is indeed thrown * update test * fix the word2vec's reset_from() * require explicit corpus size, epochs for train() * make all train() calls use explicit count, epochs * fix some error * fix test error
piskvorky · Mar 30, 2017 · becc6d3 · becc6d3
1 parent 99151db
commit becc6d3
Show file tree

Hide file tree

Showing 9 changed files with 543 additions and 178 deletions.
diff --git a/docs/notebooks/doc2vec-IMDB.ipynb b/docs/notebooks/doc2vec-IMDB.ipynb
@@ -600,7 +600,7 @@
     "        duration = 'na'\n",
     "        train_model.alpha, train_model.min_alpha = alpha, alpha\n",
     "        with elapsed_timer() as elapsed:\n",
-    "            train_model.train(doc_list)\n",
+    "            train_model.train(doc_list, total_examples=train_model.corpus_count, epochs=train_model.iter)\n",
     "            duration = '%.1f' % elapsed()\n",
     "            \n",
     "        # evaluate\n",