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]