cogdl.models.nn.gcnii

Module Contents

Classes

GCNIILayer

GCNII

class cogdl.models.nn.gcnii.GCNIILayer(n_channels, alpha=0.1, beta=1, residual=False)[source]

Bases: torch.nn.Module

reset_parameters(self)[source]
forward(self, x, edge_index, edge_attr, init_x)[source]
class cogdl.models.nn.gcnii.GCNII(in_feats, hidden_size, out_feats, num_layers, dropout=0.5, alpha=0.1, lmbda=1, wd1=0.0, wd2=0.0)[source]

Bases: cogdl.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, edge_index, edge_attr=None)[source]
loss(self, data)[source]
predict(self, data)[source]
get_optimizer(self, args)[source]