ggfm.modelsΒΆ
The Heterogeneous Graph Transformer (HGT) operator from the "Heterogeneous Graph Transformer" paper. |
|
Message Passing Neural Network (MPNN) layer :param in_channels: Number of input features :type in_channels: int :param hidden_channels: Number of hidden features :type hidden_channels: int :param out_channels: Number of output features :type out_channels: int |
|
"GPT-GNN: Generative Pre-Training of Graph Neural Networks" paper. |
|
Matching between a pair of nodes to conduct link prediction. |
|
Container module with an encoder, a recurrent module, and a decoder. |
|
Classifier for graph node classification task. |
|
LinkPredictor for graph link prediction task. |
|
CLIP model class |
|
Graph Llama model |
|
Graph Llama model for causal language modeling |
|
Graph GPT model |
|
PT-HGNN the model for heterogeneous graph neural network with pre-training. |
|
SGFormer model from the `"SGFormer: Spatial Graph Transformer for Molecular Property Prediction" :param in_channels: Number of input features. :type in_channels: int :param hidden_channels: Number of hidden features. :type hidden_channels: int :param out_channels: Number of output :type out_channels: int. |
|
HeteroLLaMA model for causal language modeling with heterogeneous graphs. |