node2vec应用记录

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也是可以的,方便查看

觉得有用的话点个赞咯~~~~~~~

原文地址:https://www.cnblogs.com/kjkj/p/10069831.html