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 |
---|---|
Default configuration for the Mambular model. |
FTTransformer#
Dataclass |
Description |
---|---|
Default configuration for the FTTransformer model. |
ResNet#
Dataclass |
Description |
---|---|
Default configuration for the ResNet model. |
MLP#
Dataclass |
Description |
---|---|
Default configuration for the MLP model. |
TabTransformer#
Dataclass |
Description |
---|---|
Default configuration for the TabTransformer model. |
MambaTab#
Dataclass |
Description |
---|---|
Default configuration for the MambaTab model. |
RNN#
Dataclass |
Description |
---|---|
Default configuration for RNN models (LSTM, GRU). |
MambAttention#
Dataclass |
Description |
---|---|
Default configuration for the MambAttention model. |
NDTF#
Dataclass |
Description |
---|---|
Default configuration for the Neural Decision Tree Forest (NDTF) model. |
NODE#
Dataclass |
Description |
---|---|
Default configuration for the Neural Oblivious Decision Ensembles (NODE) model. |
TabM#
Dataclass |
Description |
---|---|
Default configuration for the TabM model (Batch-Ensembling MLP). |
SAINT#
Dataclass |
Description |
---|---|
Default configuration for the SAINT model. |