search

dbusbridge / gcn_tutorial

A tutorial on Graph Convolutional Neural Networks
315 stars 78 forks source link

TF2.2 Port #7

Open adimyth opened 4 years ago

adimyth commented 4 years ago

Thanks for creating the repo. It allowed me to understand the basics of GCNs, a good starting point. I wanted to learn by building it from scratch, so I have ported the repo to work with TF 2.2.

  1. I have modified your following scripts to work with TF 2.2
    • layers/graph.py
    • karate_supervised.py
    • karate_unsupervised.py
  2. Used poetry for virtualenv creation, as the README says, you wished to add support for alternatives. So might be worth considering.
  3. Updated README with the required steps to setup locally

Feel free to test out the implementation & see if they work as expected. If you find any issues, kindly add a comment.