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也是可以的,方便查看
覺得有用的話點個贊咯~~~~~~~