mtenn.readout.Readout

class mtenn.readout.Readout(*args, **kwargs)[source]

Bases: Module, ABC

Abstract base class for the Readout block. Any subclass needs to implement the forward method in order to be used.

Methods

forward(delta_g)

For any readout class, this function should take the predicted \(\mathrm{\Delta G}\) value as input, and return whatever transformed value the class is implementing.

abstract forward(delta_g)[source]: For any readout class, this function should take the predicted \(\mathrm{\Delta G}\) value as input, and return whatever transformed value the class is implementing.