创建Word2Vec模型: TypeError: init() got an unexpected keyword argument ‘size‘

最新推荐文章于 2024-01-03 20:17:38 发布

原创

最新推荐文章于 2024-01-03 20:17:38 发布 · 2k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #keras #word2vec

本文档探讨了在使用gensim库创建Word2Vec模型时遇到的TypeError，原因在于参数名不匹配。作者指出了旧版本中使用'size'，而在新版中需改为'vector_size'。通过修正这一问题，读者可以顺利构建Word2Vec模型进行文本嵌入分析。

---------------------------------------------------------------------------
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'

我的代码：