报错描述

我在使用Doc2vec训练模型时，遇到了这一报错，相应的代码及报错信息如下所示：

#代码

model = Doc2Vec(x_train, min_count=5, window=5, size=size, sample=1e-4, negative=5,
                    workers=multiprocessing.cpu_count())


#报错信息

TypeError: __init__() got an unexpected keyword argument 'size'

解决办法

就是将Doc2Vec函数参数的size修改成vector_size，即修改成如下代码：

# 修改后的代码

model = Doc2Vec(x_train, min_count=5, window=5, vector_size=size, sample=1e-4, negative=5,
                     workers=multiprocessing.cpu_count())