cogdl.models.nn.pyg_unsup_graphsage
¶
Module Contents¶
Classes¶
-
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
-
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