1.已有写好的python代码,可以直接下载调用,GitHub链接https://github.com/aditya-grover/node2vec/blob/master/requirements.txt
2.代码是Python2版本,可以自己修改代码或者通过2to3.py将代码自动转换(转换方式:命令行下输入python 2to3.py -w 代码所在绝对路径)
3.转换之后还有两个地方需要改动,在main函数中:
def learn_embeddings(walks):
''' Learn embeddings by optimizing the Skipgram objective using SGD. '''
walks = [list(map(str, walk)) for walk in walks]#第一个修改的地方
model = Word2Vec(walks, size=args.dimensions, window=args.window_size, min_count=0, sg=1, workers=args.workers, iter=args.iter)
model.wv.save_word2vec_format(args.output)#第二个修改的地方
return
GitHub上提示第二个地方改为model.wv.save_word2vec_format,但是我执行的过程中发现并没有写入执行结果,所以改成了model.wv.save_word2vec_format(args.output)没有问题,运行成功。
4.源代码运行结果报存的文件格式是emb我尝试报存为xlsx也是可以的,方便查看
觉得有用的话点个赞咯~~~~~~~