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

 

 

 

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


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM