cogdl.models.nn.disengcn

Module Contents

Classes

DisenGCNLayer

Implementation of “Disentangled Graph Convolutional Networks” <http://proceedings.mlr.press/v97/ma19a.html>.

DisenGCN

class cogdl.models.nn.disengcn.DisenGCNLayer(in_feats, out_feats, K, iterations, tau=1.0, activation='leaky_relu')[source]

Bases: torch.nn.Module

Implementation of “Disentangled Graph Convolutional Networks” <http://proceedings.mlr.press/v97/ma19a.html>.

reset_parameters(self)[source]
forward(self, x, edge_index)[source]
class cogdl.models.nn.disengcn.DisenGCN(in_feats, hidden_size, num_classes, K, iterations, tau, dropout, activation)[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.

reset_parameters(self)[source]
forward(self, x, edge_index)[source]
loss(self, data)[source]
predict(self, data)[source]