其他分享
首页 > 其他分享> > gensim Word2vec转移学习(来自非gensim模型)

gensim Word2vec转移学习(来自非gensim模型)

作者:互联网

我有一组用神经网络训练的嵌入,与gensim的word2vec无关.

我想将这些嵌入用作gensim.Word2vec中的初始权重.

现在,我所看到的是可以对model.load(SOME_MODEL)进行建模,然后继续训练,但这需要使用gensim模型作为输入.同样reset_from()似乎只接受其他gensim模型.
但就我而言,我没有一个gensim模型,而是一个word2vec嵌入格式的文本文件.

那么,如何开始将学习内容从word2vec文本文件转移到gensim.Word2vec?

解决方法:

您可以使用原始Word2Vec模型的键向量格式加载其他模型

import io

from gensim.models.keyedvectors import KeyedVectors

# first line is vocab size and vector dim
model_buf = io.StringIO("""
2 3
word0 -0.000737 -0.002106 0.001851
word1 -0.000878 -0.002106 0.002834
""".lstrip())

model = KeyedVectors.load_word2vec_format(model_buf, binary=False)

model['word0']

标签:gensim,word2vec,python
来源: https://codeday.me/bug/20191013/1908690.html