relbench icon indicating copy to clipboard operation
relbench copied to clipboard

Enable use of GAT architecture in Relational Deep Learning

Open cpondoc opened this issue 11 months ago • 1 comments

As part of some exploration, we adapted the HeteroGraphSAGE layer in RelBench to create a model that uses Graph Attention Networks as opposed to GraphSAGE. Users can then choose to build their RDL pipelines using this GNN architecture for ablations on different datasets.

cc @JosephGuman @merceod

cpondoc avatar Dec 12 '24 23:12 cpondoc

Hi, for this to be merged, can you please attach experimental results comparing the existing model with the GAT model you implemented. Thanks

rishabh-ranjan avatar Dec 19 '24 06:12 rishabh-ranjan