models.nn.pyg_infomax

Module Contents

Classes

Encoder
Infomax

Functions

corruption(x, edge_index)

class models.nn.pyg_infomax.Encoder(in_channels, hidden_channels)

Bases: torch.nn.Module

forward(self, x, edge_index)

models.nn.pyg_infomax.corruption(x, edge_index)

class models.nn.pyg_infomax.Infomax(num_features, num_classes, hidden_size)

Bases: models.BaseModel

static add_args(parser)

Add model-specific arguments to the parser.

classmethod build_model_from_args(cls, args)

Build a new model instance.

forward(self, x, edge_index)

loss(self, data)

predict(self, data)