search

tensorflow / neural-structured-learning

Training neural models with structured signals.
https://www.tensorflow.org/neural_structured_learning
Apache License 2.0
980 stars 189 forks source link

[GAM] Added support for GAM + Graph Convolution Networks. #34

Closed otiliastr closed 4 years ago

otiliastr commented 4 years ago

Added support for Graph Convolution Networks (GCN) as a base classification model in Graph Agreement Models (GAM).

To support this, this pull request introduces the following differences:

  1. The data pipeline needed to be modified to contain the extra information required by GCN.
  2. This implementation of GCN feeds data in a different way than previous models, so in order to keep the previous trainers clean and easy to read, this pull request adds a separated trainer for GCN.
  3. To account for the way data is fed into GCN, we also have a separate implementation for adversarial training with GCNs, found in adversarial_sparse.py.

@kvis-google @dattias