models.nn.gcn

Module Contents

Classes

GraphConvolution

Simple GCN layer, similar to https://arxiv.org/abs/1609.02907

TKipfGCN

class models.nn.gcn.GraphConvolution(in_features, out_features, bias=True)[source]

Bases: torch.nn.Module

Simple GCN layer, similar to https://arxiv.org/abs/1609.02907

reset_parameters(self)[source]
forward(self, input, edge_index)[source]
__repr__(self)[source]
class models.nn.gcn.TKipfGCN(nfeat, nhid, nclass, dropout)[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, x, adj)[source]
loss(self, data)[source]
predict(self, data)[source]