Models

This module provides classes for the Mambular models that adhere to scikit-learn’s BaseEstimator interface.

Mambular

Modules	Description
MambularClassifier	Multi-class and binary classification tasks with a sequential Mambular Model.
MambularRegressor	Regression tasks with a sequential Mambular Model.
MambularLSS	Various statistical distribution families for different types of regression and classification tasks.

FTTransformer

Modules	Description
FTTransformerClassifier	FT transformer for classification tasks.
FTTransformerRegressor	FT transformer for regression tasks.
FTTransformerLSS	Various statistical distribution families for different types of regression and classification tasks.

MLP Models

Modules	Description
MLPClassifier	Multi-class and binary classification tasks.
MLPRegressor	MLP for regression tasks.
MLPLSS	Various statistical distribution families for different types of regression and classification tasks.

TabTransformer

Modules	Description
TabTransformerClassifier	TabTransformer for classification tasks.
TabTransformerRegressor	TabTransformer for regression tasks.
TabTransformerLSS	TabTransformer for distributional tasks.

ResNet

Modules	Description
ResNetClassifier	Multi-class and binary classification tasks using ResNet.
ResNetRegressor	Regression tasks using ResNet.
ResNetLSS	Distributional tasks using ResNet.

MambaTab

Modules	Description
MambaTabClassifier	Multi-class and binary classification tasks using MambaTab.
MambaTabRegressor	Regression tasks using MambaTab.
MambaTabLSS	Distributional tasks using MambaTab.

MambaAttention

Modules	Description
MambAttentionClassifier	Multi-class and binary classification tasks using a Combination between Mamba and Attention layers.
MambAttentionRegressor	Regression tasks using sing a Combination between Mamba and Attention layers.
MambAttentionLSS	Distributional tasks using sing a Combination between Mamba and Attention layers.

RNN Models Including LSTM and GRU

Modules	Description
TabulaRNNClassifier	Multi-class and binary classification tasks using a RNN.
TabulaRNNRegressor	Regression tasks using a RNN.
TabulaRNNLSS	Distributional tasks using a RNN.

TabM

Modules	Description
TabMClassifier	Multi-class and binary classification tasks using TabM - Batch Ensembling MLP.
TabMRegressor	Regression tasks using TabM - Batch Ensembling MLP.
TabMLSS	Distributional tasks using TabM - Batch Ensembling MLP.

NODE

Modules	Description
NODEClassifier	Multi-class and binary classification tasks using Neural Oblivious Decision Ensembles.
NODERegressor	Regression tasks using Neural Oblivious Decision Ensembles.
NODELSS	Distributional tasks using Neural Oblivious Decision Ensembles.

NDTF

Modules	Description
NDTFClassifier	Multi-class and binary classification tasks using a Neural Decision Forest.
NDTFRegressor	Regression tasks using a Neural Decision Forest
NDTFLSS	Distributional tasks using a Neural Decision Forest.

SAINT

Modules	Description
SAINTClassifier	Multi-class and binary classification tasks using SAINT.
SAINTRegressor	Regression tasks using SAINT.
SAINTLSS	Distributional tasks using SAINT.

Base Classes

Modules	Description
SklearnBaseClassifier	Base class for classification tasks.
SklearnBaseLSS	Base class for distributional tasks.
SklearnBaseRegressor	Base class for regression tasks.

• mambular.models