openhgnn.models.HMPNN

class HMPNN(in_dim, hid_dim, out_dim, etypes, num_layers, device)[source]

Heterogeneous Message-Passing Neural Network (HMPNN)

A simple implementation of HMPNN from paper for experimenting in KTN.

Parameters:

  • in_dim (int) – Input feature size.

  • hid_dim (int) – Hidden layer size.

  • out_dim (int) – Output feature size.

  • etypes (list) – Edge types.

  • num_layers (int) – Number of layers.

  • device (str) – Device to run the model.