JK-Net-Pytorch

Pytorch implementation of JK-Net, paper: Representation Learning on Graphs with Jumping Knowledge Networks. Supported datasets are cora and citeseer.

Referenced Repositories:

Requirements

* PyTorch 1.14
* Python 3.7

python train.py --dataset dataset_name --model model_name --layer layer_name

--dataset_name: core, citseer
--model: Concat, Maxpooling, LSTM, GCN, GAT
--layer: GCN, GAT

Paper: Representation Learning on Graphs with Jumping Knowledge Networks

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
data		data
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
layer.py		layer.py
model.py		model.py
train.py		train.py
utils.py		utils.py