zoukankan      html  css  js  c++  java
  • 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也是可以的,方便查看

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

  • 相关阅读:
    最小生成树
    负环详解
    P2053 [SCOI2007]修车
    P3254 圆桌问题
    P3114 [USACO15JAN]踩踏Stampede
    SP1043 GSS1
    SP2713 GSS4
    导出mysql内数据 python建倒排索引
    社团管理系统——总结报告
    北京地铁出行线路规划——代码实现
  • 原文地址:https://www.cnblogs.com/kjkj/p/10069831.html
Copyright © 2011-2022 走看看