torchcell.models.hetero_cell_nsa_retry.PreProcessor

class PreProcessor(in_channels: int, hidden_channels: int, num_layers: int = 2, dropout: float = 0.1, norm: str = 'layer', activation: str = 'relu')

Bases: Module

MLP for preprocessing node features.

forward(x: Tensor) → Tensor

Pass node features through the preprocessing MLP.