magnet.aggmodels.gnn

Classes

GNN(*args, **kwargs)	Abstract base class for Graph Neural networks for mesh agglomeration.
GeometricGNN(*args, **kwargs)
ReinforceLearnGNN(*args, **kwargs)	Reinforcement Learning GNN for graph partitioning.