magnet.aggmodels.sageheterogeneous

Classes

GNNHeterogeneous(*args, **kwargs)	Abstract base class for GNNs for agglomerating heterogeneous meshes.
SageHeterogeneous(hidden_units, ...[, ...])	GNN with 4 SAGE convolutional layers and 4 linear layers.