openhgnn.models.HGT

class HGT(in_dim, out_dim, num_heads, num_etypes, ntypes, num_layers, dropout=0.2, norm=False)[source]

Heterogeneous graph transformer convolution from Heterogeneous Graph Transformer

For more details, you may refer to `HGT<https://docs.dgl.ai/en/0.8.x/generated/dgl.nn.pytorch.conv.HGTConv.html>`__

Parameters:

  • in_dim (int) – the input dimension

  • out_dim (int) – the output dimension

  • num_heads (list) – the list of the number of heads in each layer

  • num_etypes (int) – the number of the edge type

  • num_ntypes (int) – the number of the node type

  • num_layers (int) – the number of layers we used in the computing

  • dropout (float) – the feature drop rate

  • norm (boolean) – if we need the norm operation

  • ntypes (list) – the list of node type