cogdl.models.nn.pyg_gtn

Module Contents

Classes

GTConv

GTLayer

GTN

class cogdl.models.nn.pyg_gtn.GTConv(in_channels, out_channels, num_nodes)[source]

Bases: torch.nn.Module

reset_parameters(self)[source]
forward(self, A)[source]
class cogdl.models.nn.pyg_gtn.GTLayer(in_channels, out_channels, num_nodes, first=True)[source]

Bases: torch.nn.Module

forward(self, A, H_=None)[source]
class cogdl.models.nn.pyg_gtn.GTN(num_edge, num_channels, w_in, w_out, num_class, num_nodes, num_layers)[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.

normalization(self, H)[source]
norm(self, edge_index, num_nodes, edge_weight, improved=False, dtype=None)[source]
forward(self, A, X, target_x, target)[source]
loss(self, data)[source]
evaluate(self, data, nodes, targets)[source]