dhg.nn
Common Layers
A Multi-Layer Perception (MLP) model. |
|
A wrapper to apply multiple heads to a given layer. |
|
The Discriminator for Generative Adversarial Networks (GANs). |
Layers on Graph
The GCN convolution layer proposed in Semi-Supervised Classification with Graph Convolutional Networks paper (ICLR 2017). |
|
The GraphSAGE convolution layer proposed in Inductive Representation Learning on Large Graphs paper (NeurIPS 2017). |
|
The GAT convolution layer proposed in Graph Attention Networks paper (ICLR 2018). |
|
The GIN convolution layer proposed in How Powerful are Graph Neural Networks? paper (ICLR 2019). |
Layers on Hypergraph
The HGNN convolution layer proposed in Hypergraph Neural Networks paper (AAAI 2019). |
|
The HGNN + convolution layer proposed in HGNN+: General Hypergraph Neural Networks paper (IEEE T-PAMI 2022). |
|
The Jump Hypergraph Convolution layer proposed in Dual Channel Hypergraph Collaborative Filtering paper (KDD 2020). |
|
The HyperGCN convolution layer proposed in HyperGCN: A New Method of Training Graph Convolutional Networks on Hypergraphs paper (NeurIPS 2019). |
|
The HNHN convolution layer proposed in HNHN: Hypergraph Networks with Hyperedge Neurons paper (ICML 2020). |
|
The UniGCN convolution layer proposed in UniGNN: a Unified Framework for Graph and Hypergraph Neural Networks paper (IJCAI 2021). |
|
The UniGAT convolution layer proposed in UniGNN: a Unified Framework for Graph and Hypergraph Neural Networks paper (IJCAI 2021). |
|
The UniSAGE convolution layer proposed in UniGNN: a Unified Framework for Graph and Hypergraph Neural Networks paper (IJCAI 2021). |
|
The UniGIN convolution layer proposed in UniGNN: a Unified Framework for Graph and Hypergraph Neural Networks paper (IJCAI 2021). |
Loss Functions
This criterion computes the Bayesian Personalized Ranking (BPR) loss between the positive scores and the negative scores. |
Regularizations
Regularization function for embeddings. |