diff options
Diffstat (limited to 'tensorflow/models/embedding/word2vec.py')
-rw-r--r-- | tensorflow/models/embedding/word2vec.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/models/embedding/word2vec.py b/tensorflow/models/embedding/word2vec.py index b8afbd2ced..97b38b7f47 100644 --- a/tensorflow/models/embedding/word2vec.py +++ b/tensorflow/models/embedding/word2vec.py @@ -378,7 +378,8 @@ class Word2Vec(object): opts = self._options with open(os.path.join(opts.save_path, "vocab.txt"), "w") as f: for i in xrange(opts.vocab_size): - f.write("%s %d\n" % (tf.compat.as_text(opts.vocab_words[i]), + vocab_word = tf.compat.as_text(opts.vocab_words[i]).encode("utf-8") + f.write("%s %d\n" % (vocab_word, opts.vocab_counts[i])) def _train_thread_body(self): |