bioneuralnet.network_embedding.gnn_models

Classes

GAT(*args, **kwargs)	GAT with optional final regression layer.
GCN(*args, **kwargs)	GCN with optional final regression layer.
GIN(*args, **kwargs)	GIN with optional final regression layer.
SAGE(*args, **kwargs)	GraphSAGE with optional final regression layer.