openhgnn.layers.HeteroEmbedLayer

class HeteroEmbedLayer(n_nodes_dict, embed_size, embed_name='embed', activation=None, dropout=0.0)

Embedding layer for featureless heterograph.

Parameters:

• n_nodes_dict (dict[str, int]) – Key of dict means node type, value of dict means number of nodes.

• embed_size (int) – Dimension of embedding,

• activation (callable activation function/layer or None, optional) – If not None, applies an activation function to the updated node features. Default: None.

• dropout (float, optional) – Dropout rate. Default: 0.0