Write UTF-8 byte strings in tensorboard conversion (#1144)

piskvorky · Feb 14, 2017 · 923c656 · 923c656
1 parent cebee95
commit 923c656
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/scripts/word2vec2tensor.py b/gensim/scripts/word2vec2tensor.py
@@ -51,7 +51,7 @@ def word2vec2tensor(word2vec_model_path,tensor_filename, binary=False):
     with open(outfiletsv, 'w+') as file_vector:
         with open(outfiletsvmeta, 'w+') as file_metadata:
             for word in model.index2word:
-                file_metadata.write(gensim.utils.to_unicode(word) + '\n')
+                file_metadata.write(gensim.utils.to_utf8(word) + gensim.utils.to_utf8('\n'))
                 vector_row = '\t'.join(map(str, model[word]))
                 file_vector.write(vector_row + '\n')