magnet.aggmodels.sagebase

Classes

SageBase(hidden_units, lin_hidden_units, ...)	GNN with 4 SAGE convolutional layers and 4 linear layers.
SageBase2D([hidden_units, lin_hidden_units, ...])	GNN with 4 SAGE convolutional layers and 3 linear layers.
SageRes()	GNN that uses residual connections.