NRL实施
重新实现四种网络表示学习(NRL)算法:DeepWalk,LINE,node2vec,GraphGAN。
环境
NumPy
TensorFlow
Gensim
网络X
数据
路径./data/中有两个数据集:
:引文数据集。
:以下网络。
训练
首先,找到项目的根路径:
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
DeepWalk可以用LIN
2022-04-08 03:37:55
9.33MB
Python
1