---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-11-f65b2b409400> in <module>
15 for window_param in [1, 3, 5, 10]:
16 # generate model
---> 17 model = create_models(cases, size_param, window_param, 1)
18
19 # calculating the average feature vector for each sentence (trace)
<ipython-input-7-a776889d1549> in create_models(cases, size, window, min_count)
27 Creates a word2vec model
28 '''
---> 29 model = Word2Vec(size=size, window=window, min_count=min_count, workers=n_workers)
30 model.build_vocab(cases)
31 model.train(cases, total_examples=len(cases), epochs=10)
TypeError: __init__() got an unexpected keyword argument 'size'
我的代码:
def create_models(cases, size, window, min_count):
'''
Creates a word2vec model
'''
model = Word2Vec(vector_size=size, window=window, min_count=min_count, workers=n_workers)
model.build_vocab(cases)
model.train(cases, total_examples=len(cases), epochs=10)
return model
在创建word2vec模型时报了上述错误,原因是gensim库的版本问题,之前的版本是size,新版需要将其改为vector_size。根据官网说明models.word2vec – Word2vec embeddings — gensim
原来是形参名字错了,需要将size改为vector_size