134

GitHub - shagunsodhani/pregel: Tensorflow implementation of Graph Convolutional...

 6 years ago
source link: https://github.com/shagunsodhani/pregel
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

pregel

Tensorflow implementation of Graph Convolutional Network

Setup

  • sudo pip3 install -r requirements.txt
  • python3 main.py -h to view all the config parameters. Update the default parameters in the main.py file.
  • python3 main.py to run the models.
  • tensorboard --logdir=run1:<tensorboard-dir> -port 6006 to run tensorboard and go to http://localhost/6006

References

This work is an attempt to reproduce some of the works related to graph convolutional networks:

Semi-Supervised Classification with Graph Convolutional Networks

@article{kipf2016semi,
  title={Semi-Supervised Classification with Graph Convolutional Networks},
  author={Kipf, Thomas N and Welling, Max},
  journal={arXiv preprint arXiv:1609.02907},
  year={2016}
}

Variational Graph Auto-Encoders

@article{hamilton2017representation,
  title={Representation Learning on Graphs: Methods and Applications},
  author={Hamilton, William L and Ying, Rex and Leskovec, Jure},
  journal={arXiv preprint arXiv:1709.05584},
  year={2017}
}

report

Recommend

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK