models.nn.pyg_srgcn

Module Contents

Classes

NodeAdaptiveEncoder

SrgcnHead

SrgcnSoftmaxHead

SRGCN

class models.nn.pyg_srgcn.NodeAdaptiveEncoder(num_features, dropout=0.5)[source]

Bases: nn.Module

forward(self, x)[source]
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

forward(self, x, edge_index, edge_attr)[source]
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

forward(self, x, edge_index, edge_attr)[source]
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

static add_args(parser)[source]

Add model-specific arguments to the parser.

classmethod build_model_from_args(cls, args)[source]

Build a new model instance.

forward(self, batch)[source]
loss(self, data)[source]
predict(self, data)[source]