cogdl.models.nn.fastgcn

Module Contents

Classes

GraphConvolution

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

FastGCN

class cogdl.models.nn.fastgcn.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, adj)[source]
__repr__(self)[source]
class cogdl.models.nn.fastgcn.FastGCN(num_features, num_classes, hidden_size, num_layers, dropout, sample_size)[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.

set_adj(self, edge_index, num_nodes)[source]
_sample_one_layer(self, sampled, sample_size)[source]
_generate_adj(self, sample1, sample2)[source]
sampling(self, x, v)[source]
forward(self, x, adj)[source]