openhgnn.layers.HeteroEmbedLayer¶
- class HeteroEmbedLayer(n_nodes_dict, embed_size, embed_name='embed', activation=None, dropout=0.0)[源代码]¶
Embedding layer for featureless heterograph.
- 参数:
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