gigl.src.common.models.pyg.heterogeneous#
Classes
Heterogeneous Graph Transformer model. Paper: https://arxiv.org/pdf/2003.01332.pdf This implementation is based on the example of: pyg-team/pytorch_geometric Args: node_type_to_feat_dim_map (Dict[NodeType, int]): Dictionary mapping node types to their input dimensions. edge_type_to_feat_dim_map (Dict[EdgeType, int]): Dictionary mapping node types to their feature dimensions. hid_dim (int): Hidden dimension size. out_dim (int, optional): Output dimension size. Defaults to 128. num_layers (int, optional): Number of layers. Defaults to 2. num_heads (int, optional): Number of attention heads. Defaults to 2. |
|