Configurations#

This module provides default configurations for Mambular models. Each configuration is implemented as a dataclass, offering a structured way to define model-specific hyperparameters.

Mambular#

Dataclass

Description

DefaultMambularConfig

Default configuration for the Mambular model.

FTTransformer#

Dataclass

Description

DefaultFTTransformerConfig

Default configuration for the FTTransformer model.

ResNet#

Dataclass

Description

DefaultResNetConfig

Default configuration for the ResNet model.

MLP#

Dataclass

Description

DefaultMLPConfig

Default configuration for the MLP model.

TabTransformer#

Dataclass

Description

DefaultTabTransformerConfig

Default configuration for the TabTransformer model.

MambaTab#

Dataclass

Description

DefaultMambaTabConfig

Default configuration for the MambaTab model.

RNN#

Dataclass

Description

DefaultTabulaRNNConfig

Default configuration for RNN models (LSTM, GRU).

MambAttention#

Dataclass

Description

DefaultMambAttentionConfig

Default configuration for the MambAttention model.

NDTF#

Dataclass

Description

DefaultNDTFConfig

Default configuration for the Neural Decision Tree Forest (NDTF) model.

NODE#

Dataclass

Description

DefaultNODEConfig

Default configuration for the Neural Oblivious Decision Ensembles (NODE) model.

TabM#

Dataclass

Description

DefaultTabMConfig

Default configuration for the TabM model (Batch-Ensembling MLP).

SAINT#

Dataclass

Description

DefaultSAINTConfig

Default configuration for the SAINT model.