openhgnn.layers.ATTConv

class ATTConv(ntypes, dim)[源代码]

It is macro_layer of the models [HetGNN]. It presents in the 3.3.2 Types Combination of the paper.

In this framework, to make embedding dimension consistent and models tuning easy, we use the same dimension d for content embedding in Section 3.2, aggregated content embedding in Section 3.3, and output node embedding in Section 3.3.

So just give one dim parameter.

参数:
  • dim (int) – Input feature dimension.

  • ntypes (list) – Node types.

  • Note

    We don’t implement multi-heads version.

    atten_w is specific to the center node type, agnostic to the neighbor node type.