graph-convolutional-networks

graph convolutional networks pytorch implementation

Installing

To install the dependencies, run:

pip install -r requirements.txt

---

add cuda 10.1 and pytorch 1.4.0

Run

Use --help to see src/train.py Usage.

  
usage: train.py [-h] [--dataset DATASET] [--seed SEED]

optional arguments:
  -h, --help         show this help message and exit
  --dataset DATASET  Add Dataset
  --seed SEED        Add Seed

Example:

  python3 src/train.py --dataset ../cora/ SEED 10

Plots loss and accuracy

[train val accuracy]

[train val loss]

---