mtenn.strategy

Implementations for the Strategy block in a Model or GroupedModel.

Classes

ComplexOnlyStrategy	Strategy to only predict based on the complex representation.
ConcatStrategy	Strategy for combining the complex representation and parts representations in some learned manner, using sum-pooling to ensure permutation-invariance of the parts.
DeltaStrategy	Simple strategy for subtracting the sum of the individual component energies from the complex energy.
SplitConcatStrategy	Strategy for combining the complex representation and parts representations in some learned manner.
SplitDeltaStrategy	Simple strategy for subtracting the sum of the individual component energies from the complex energy.
Strategy	Abstract base class for the Strategy block.