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