类型错误:__init__() 得到意外的关键字参数“size”

yin*_*nky 6 python stellargraph

这是我下面的代码,我遇到的错误就在它下面,但我无法弄清楚为什么会发生这种情况。请分享您的想法

from gensim.models import word2vec
np.set_printoptions(suppress=True)

feature_size = 150
context_size= 2
min_word = 1
word_vec= word2vec.Word2Vec(tokenized, size=feature_size, \
                            window=context_size, min_count=min_word, \
                            iter=50, seed=42)



---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-59-dbe3a4fa3884> in <module>
      5 context_size= 2
      6 min_word = 1
----> 7 word_vec= word2vec.Word2Vec(tokenized, size=feature_size, \
      8                             window=context_size, min_count=min_word, \
      9                             iter=50, seed=42)

TypeError: __init__() got an unexpected keyword argument 'size'
Run Code Online (Sandbox Code Playgroud)

小智 15

我遇到了同样的问题,并通过查找Word2Vec嵌入文档解决了它。请注意,新 Gensim 中的参数有两处变化:

[1] size -> vector_size
[2] iter -> epochs
Run Code Online (Sandbox Code Playgroud)

这是文档中的代码示例:

from gensim.test.utils import common_texts
from gensim.models import Word2Vec

model = Word2Vec(sentences=common_texts, vector_size=100, window=5, min_count=1, workers=4)
model.save("word2vec.model")
model = Word2Vec.load("word2vec.model")
model.train([["hello", "world"]], total_examples=1, epochs=1)
Run Code Online (Sandbox Code Playgroud)