gammagl.models.HGTModel

class HGTModel(data, hidden_channels, out_channels, num_heads, num_layers, target_node_type, drop_rate=0.5)

The Heterogeneous Graph Transformer (HGT) proposed in here paper.

Parameters:

cfg (configuration of HGT model.) –

forward(x_dict, edge_index_dict)