models.nn.pyg_srgcn
¶
Module Contents¶
Classes¶
-
class
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:
nn.Module
-
class
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:
nn.Module
-
class
models.nn.pyg_srgcn.
SRGCN
(num_features, hidden_size, num_classes, attention, activation, nhop, normalization, dropout, node_dropout, alpha, nhead, subheads)[source]¶ Bases:
models.BaseModel