model = Word2Vec(**kwargs)时出现__init__() got an unexpected keyword argument ‘size‘

问题描述

运行网上提供的Node2Vec代码，在下面的model.train这一行提示出现了错误

if __name__ == "__main__":
    G=nx.read_edgelist('../data/wiki/Wiki_edgelist.txt',
                         create_using = nx.DiGraph(), nodetype = None, data = [('weight', int)])
    model = Node2Vec(G, walk_length=10, num_walks=80,
                     p=0.25, q=4, workers=1, use_rejection_sampling=0)
    model.train(window_size = 5, iter = 3)
    embeddings = model.get_embeddings()

    evaluate_embeddings(embeddings)
    plot_embeddings(embeddings)

错误提示如下
Traceback (most recent call last):
File “D:\desktop\GraphEmbedding-master\examples\node2vec_wiki.py”, line 49, in
model.train(window_size = 5, iter = 3)
File “D:\desktop\GraphEmbedding-master\ge\models\node2vec.py”, line 54, in train
model = Word2Vec(**kwargs)
TypeError: init() got an unexpected keyword argument ‘size’

根据提示发现是在使用Nove2Vec的train对象时，出现了参数上的错误

问题分析

这是错误提示
点击红框这一部分，会跳到下图位置，我们得知是Word2Vec上传的参数出现了问题，即类似于
kwargs[“sentences”] = self.sentences
kwargs[“min_count”] = kwargs.get(“min_count”, 0)
kwargs[“size”] = embed_size
一类的出现了问题，也即是""里面的参数名字不对
那么我们需要知道Word2Vec的定义，才能知道具体要改成什么样的参数
双击“shift”，进入pycharm的搜索界面，输入Word2Vec，找到了它的定义
点击后，进入Word2Vec的定义界面，看到了函数参数的形式

解决方案

现在我们把两部分拿来对比，将上面红框里，双引号内的参数，改为和下面的参数保持一致
分别有：①size改为vector_size②iter改为epochs
改完后就可以运行了。不同的代码可能不一样， 关于Word2Vec(**kwargs)的问题大多是这个原因，根据上面的流程操作即可。大家有什么疑问欢迎留言讨论