
NRL-Implement: 网络表示学习方法实现,涵盖DeepWalk、LINE、node2vec及GraphGAN等算法(基于TensorFlow)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
NRL-Implement是一个基于TensorFlow构建的网络表示学习库,实现了多种经典算法如DeepWalk、LINE和node2vec,并新增了GraphGAN。
NRL实施重新实现了四种网络表示学习(NRL)算法:DeepWalk、LINE、node2vec以及GraphGAN。所需环境包括NumPy、TensorFlow、Gensim及NetworkX。数据路径为./data/,其中包含两个数据集:
1. 引文数据集。
2. 其它网络。
训练步骤如下:
首先找到项目的根路径:`cd NRL-implement`
对于DeepWalk的执行命令是:`python DeepWalk/main.py`
对于LINE的执行命令是:`python LINE/main.py`
对于node2vec的执行命令是:`python node2vec/main.py`
这三个实现使用cora作为数据集,结果保存在./results/cora/中。 使用逻辑回归作为分类器来评估这三种方法产生的嵌入质量,其对应的命令为: `python LRclassifier.py --method DeepWalk`
全部评论 (0)
还没有任何评论哟~


