diff options
author | 2016-07-25 13:48:16 -0800 | |
---|---|---|
committer | 2016-07-25 15:02:34 -0700 | |
commit | 21716d8f6e175cd6e8cd97a84e48497574268b0c (patch) | |
tree | 3345202e7a812cc9572beb24fc01732696a4140d /tensorflow/models/embedding/word2vec.py | |
parent | ed281973d66d0030e58a77a05821bbb88627f5bd (diff) |
Merge changes from github.
Change: 128401884
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): |