acbull/pyHGT
A graph neural network architecture for learning on large-scale heterogeneous and dynamic graphs, published at WWW 2020.

This repository implements the Heterogeneous Graph Transformer (HGT), a GNN architecture designed to handle heterogeneous graphs with multiple node and edge types. The model adapts the transformer mechanism to graph-structured data, using type-specific parameters and attention to capture relations across different entity types. It provides sampling algorithms for scalable subgraph extraction and training scripts for downstream tasks like node classification and link prediction.