/gin-attn-conv-pytorch

Unofficial implementation of Graph Isomorphism Network in Python

Primary LanguagePythonMIT LicenseMIT

gin_attn_conv_pytorch

Experiment combining Graph Attention Network (GAT) and Graph Isomorphism Network (GIN) to create the GINAttnConv layer.

Running

Run the following command to compare GCN, GAT, and GINAttnConv models:

$ python main.py