graphnet.models.gnn.gnn module

Base GNN-specific Model class(es).

class graphnet.models.gnn.gnn.GNN(*args, **kwargs)

Bases: Model

Base class for all core GNN models in graphnet.

Construct GNN.

Parameters:

• args (Any)

• kwargs (Any)

Return type:

object

property nb_inputs: int

Return number of input features.

property nb_outputs: int

Return number of output features.

abstract forward(data)

Apply learnable forward pass in model.

Return type:

Tensor

Parameters:

data (Data)