將 Word2Vec 從 dl4j 導出到 vector.tsv 和 metadata.tsv 用於 tensorflow 嵌入投影儀？

Question

我已經使用 deeplearning4j 訓練了 Word2Vec 嵌入。

如何將生成的Word2Vec model 導出到vectors.tsv和metadata.tsv以便我可以使用https://projector.tensorflow.org對其進行可視化

Answer 1

試試這個。

# your word2vec model loaded in `w2v_model`
with open('tensors.tsv', 'w') as tensors:
    with open('metadata.tsv', 'w') as metadata:
        for word in w2v_model.wv.index2word:
            metadata.write(word + '\n')
            vector_row = '\t'.join(map(str, w2v_model.wv[word]))
            tensors.write(vector_row + '\n')