Skip to content

renH2/GCN-GateGCN-model

Repository files navigation

utils_load:处理数据所需函数

GNN_model:目前只有MLP,两种GCN(GCN_Module、GcnNet),多层GCN (GatedGNN还有点问题)


运行方法:运行GNN_model.py即可


修改数据集:(目前可选cora和citeseer,pubmed找不到...)

image-20201118214633671

加载数据集

image-20201118214748795

f为feature matrix,维数为[node_num, feature_num]

a为邻接矩阵,l为label矩阵

train_idx,valid_idx,test_idx为训练、验证、测试的掩码

超参数修改

image-20201118215052633

模型选择:

image-20201118215215313

About

GCN&GateGCN model on cora&citeseer by Pytorch 🐳

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages