gcn

Architectures for convolutional layers

Classes

BaseGCNStack

A wrapper around a stack of GCN graph convolutional layers.

SAGEConvStack

GraphSAGE graph convolutional neural network for atom embeddings.

GINConvStack

Graph Isomorphism Network GCN for whole molecule embeddings.