cogdl.models.nn.pyg_unsup_graphsage

Module Contents

Classes

SAGE

Graphsage

class cogdl.models.nn.pyg_unsup_graphsage.SAGE(num_features, hidden_size, num_layers, sample_size, dropout, walk_length, negative_samples)[source]

Bases: torch.nn.Module

sampling(self, edge_index, num_sample)[source]
forward(self, x, edge_index)[source]
loss(self, data)[source]
embed(self, data)[source]
class cogdl.models.nn.pyg_unsup_graphsage.Graphsage(num_features, hidden_size, num_classes, num_layers, sample_size, dropout, walk_length, negative_samples, lr, epochs, patience)[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.

train(self, data)[source]