如何手动将单词和矢量添加到Word2vec gensim



比方说,word2vec.model是我训练过的word2vec模型。当出现词汇表外单词(oov_word(时,我使用compute_vec(oov_wword(方法计算向量vec。现在,我想将oov_word及其相应的向量vec添加到我已经训练好的模型word2vec.model中。

我已经检查了以下链接。但他们没有回答我的问题。

组合/添加来自不同word2vec模型的矢量

https://datascience.stackexchange.com/questions/49431/how-to-train-an-existing-word2vec-gensim-model-on-new-words

https://radimrehurek.com/gensim/models/keyedvectors.html#gensim.models.keyedvectors.BaseKeyedVectors.add

from gensim.models.keyedvectors import WordEmbeddingsKeyedVectors
vector_length = 100
kv = WordEmbeddingsKeyedVectors(vector_length)
# wordList - list of words
# vectorList - list of the vector corresponding to the words
kv.add(wordList, vectorList)
kv.most_similar(word1) # gives the list of words similar to word1

4.0版:

w2v_model.wv.add_vectors(keys, vectors)

检查文档

最新更新