项目作者: zbyzby11

项目描述 :
一些graph/network embedding算法的pytorch实现
高级语言: Python
项目地址: git://github.com/zbyzby11/GraphEmbedding.git
创建时间: 2019-10-17T05:16:42Z
项目社区:https://github.com/zbyzby11/GraphEmbedding

开源协议:

下载


1. 下载依赖

项目使用python版本为3.7

pytorch1.1版本

运行以下命令安装依赖包:

pip install -r requirements.txt

2. 算法运行

输入一个文件,edgelist格式的输入,每行表示网络中相邻节点对,格式如下:

node1 node2

node1 node3

……

由于未完成算法的统一输入函数,所以现在可以在每个分算法下独立运行算法

输出为:一个文件,每行表示节点的id和它的向量表示,格式如下:

node1 0.1 0.1 0.2 ……

node2 0.2 0.13 0.2 ……

……

独立运行每个算法文件:

python [算法名文件], 如python line.py

3. 说明

本项目是一些有关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)的算法