一些graph/network embedding算法的pytorch实现
项目使用python版本为3.7
pytorch1.1版本
运行以下命令安装依赖包:
pip install -r requirements.txt
输入一个文件,edgelist格式的输入,每行表示网络中相邻节点对,格式如下:
node1 node2
node1 node3
……
由于未完成算法的统一输入函数,所以现在可以在每个分算法下独立运行算法
输出为:一个文件,每行表示节点的id和它的向量表示,格式如下:
node1 0.1 0.1 0.2 ……
node2 0.2 0.13 0.2 ……
……
独立运行每个算法文件:
python [算法名文件], 如python line.py
本项目是一些有关network embedding和GNN在一些数据及上的算法实现,更新的算法会在后续添加
10-17 已经完成deepwalk算法
10-21 已经完成node2vec算法
10-23 已经完成line算法
================================
需要进行训练好的embedding的效果测试
10-25 上述三种模型经过测试,效果良好
================================
下一步继续编写算法sdne
=======
下一步继续编写算法sdne与GCN
10-30 完成对SDNE算法的调试与测试
11-2 完成GCN算法的调试与测试
开题中。。。
下一步编写出graph attention network(GAT)的算法