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

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

  • 5
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值