GINConvStack

class openff.nagl.nn.gcn.GINConvStack(*args, _is_dgl: bool = False, **kwargs)[source]

Bases: BaseGCNStack[GINConv]

Graph Isomorphism Network GCN for whole molecule embeddings.

Attributes

available_aggregator_types

default_activation_function

default_aggregator_type

default_dropout

name

training

available_aggregator_types = ['sum', 'max', 'mean']
default_activation_function = <class 'torch.nn.modules.activation.ReLU'>
default_aggregator_type = 'sum'
default_dropout = 0.0
name = 'GINConv'
training: bool