← all repositories

acbull/pyHGT

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

922 stars Python ML Frameworks
pyHGT
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.