cogdl.models.nn.pyg_srgcn
¶
Module Contents¶
Classes¶
-
class
cogdl.models.nn.pyg_srgcn.
NodeAdaptiveEncoder
(num_features, dropout=0.5)[source]¶ Bases:
cogdl.layers.srgcn_module.nn.Module
-
class
cogdl.models.nn.pyg_srgcn.
SrgcnHead
(num_features, out_feats, attention, activation, normalization, nhop, subheads=2, dropout=0.5, node_dropout=0.5, alpha=0.2, concat=True)[source]¶ Bases:
cogdl.layers.srgcn_module.nn.Module
-
class
cogdl.models.nn.pyg_srgcn.
SrgcnSoftmaxHead
(num_features, out_feats, attention, activation, nhop, normalization, dropout=0.5, node_dropout=0.5, alpha=0.2)[source]¶ Bases:
cogdl.layers.srgcn_module.nn.Module
-
class
cogdl.models.nn.pyg_srgcn.
SRGCN
(num_features, hidden_size, num_classes, attention, activation, nhop, normalization, dropout, node_dropout, alpha, nhead, subheads)[source]¶ Bases:
cogdl.models.BaseModel