models.nn.unsup_graphsage
¶
Module Contents¶
Classes¶
-
class
models.nn.unsup_graphsage.
SAGE
(num_features, hidden_size, num_layers, sample_size, dropout, walk_length, negative_samples)[source]¶ Bases:
torch.nn.Module
-
class
models.nn.unsup_graphsage.
Graphsage
(num_features, hidden_size, num_classes, num_layers, sample_size, dropout, walk_length, negative_samples, lr, epochs)[source]¶ Bases:
models.BaseModel