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

 

 

 

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

相关文章:

  • 2021-09-27
  • 2022-12-23
  • 2021-12-31
  • 2021-10-24
  • 2021-07-29
  • 2022-03-03
  • 2021-05-27
  • 2021-11-28
猜你喜欢
  • 2021-09-10
  • 2021-09-08
  • 2021-06-11
  • 2021-06-14
  • 2021-06-23
  • 2021-11-22
  • 2021-12-20
相关资源
相似解决方案