terminator.models.layers.term.graph.s2s

TERM MPNN modules

This file contains Attention and Message Passing implementations of the TERM MPNN.

Classes

TERMEdgeEndpointAttention(num_hidden, num_in)

TERM Edge Endpoint Attention

TERMEdgeMPNNLayer(num_hidden, num_in[, ...])

TERM Edge MPNN Layer

TERMEdgeTransformerLayer(num_hidden, num_in)

TERM Edge Transformer Layer

TERMGraphTransformerEncoder(hparams)

TERM Graph Transformer Encoder

TERMNeighborAttention(num_hidden, num_in[, ...])

TERM Neighbor Attention

TERMNodeMPNNLayer(num_hidden, num_in[, ...])

TERM Node MPNN Layer

TERMNodeTransformerLayer(num_hidden, num_in)

TERM Node Transformer Layer