enerzyme.models.blocks.mlp.NeuronLayer

class enerzyme.models.blocks.mlp.NeuronLayer(dim_feature_in, dim_feature_out, activation_fn: Literal['shifted_softplus', 'swish'] | None = None, activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {})

Bases: Module

__init__(dim_feature_in, dim_feature_out, activation_fn: Literal['shifted_softplus', 'swish'] | None = None, activation_params: Dict[Literal['dim_feature', 'initial_alpha', 'initial_beta', 'learnable'], int | float | bool] = {}) → None

Initialize internal Module state, shared by both nn.Module and ScriptModule.