【报错解决】TypeError: init() got an unexpected keyword argument ‘size‘

小白冲鸭

于 2024-01-03 20:17:38 发布

阅读量1.1k

点赞数 13

CC 4.0 BY-SA版权

分类专栏：报错解决文章标签：深度学习机器学习人工智能自然语言处理 Doc2Vec Word2Vec

本文链接：https://blog.youkuaiyun.com/m0_56367027/article/details/135372147

报错解决专栏收录该内容

12 篇文章

订阅专栏

在使用Doc2Vec进行模型训练时遇到TypeError，原因是该库接受vector_size而非size作为向量维度参数。修正后的代码将size替换为vector_size解决了问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

报错描述

我在使用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())