ggfm.models.MPNN

class ggfm.models.MPNN(in_channels, hidden_channels, out_channels, **kwargs)

Message Passing Neural Network (MPNN) layer :param in_channels: Number of input features :type in_channels: int :param hidden_channels: Number of hidden features :type hidden_channels: int :param out_channels: Number of output features :type out_channels: int

forward(g, use_conv=True)

Forward pass of MPNN layer :param g: :param use_conv:

reset_parameters()

training: bool