openhgnn.models.RHGNN

class RHGNN(graph: DGLGraph, input_dim_dict, hidden_dim: int, relation_input_dim: int, relation_hidden_dim: int, num_layers: int, category, out_dim, n_heads: int = 4, dropout: float = 0.2, negative_slope: float = 0.2, residual: bool = True, norm: bool = True)[source]

This is the main method of model RHGNN

Parameters:

  • graph (dgl.DGLHeteroGraph) – a heterogeneous graph

  • input_dim_dict (dict) – node input dimension dictionary

  • hidden_dim (int) – node hidden dimension

  • relation_input_dim (int) – relation input dimension

  • relation_hidden_dim (int) – relation hidden dimension

  • num_layers (int) – number of stacked layers

  • n_heads (int) – number of attention heads

  • dropout (float) – dropout rate

  • negative_slope (float) – negative slope

  • residual (boolean) – residual connections or not

  • norm (boolean) – layer normalization or not