Registry / ai-ml / pytabkit

pytabkit

JSON →
library1.7.3pypypi✓ verified 82d ago

PyTabKit provides ML models and a benchmark for tabular data classification and regression. Current version is 1.7.3, supporting Python >=3.9, with PyTorch-based models like RealMLP, xRFM, TabM, and TabICL wrappers, plus scikit-learn-style interfaces, HPO, and ensembling. Release cadence is irregular, with several updates in 2024-2025.

pip install pytabkit
INSTALL
IMPORT
SIG · PYTABKIT
P
pytabkit
ai-mlpythonv1.7.3
harness data pending
Install & Compatibility
Where this runs

No compatibility data collected yet for this library.

Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

CatBoost_D_Classifier
from pytabkit import CatBoost_D_Classifier
from pytabkit.models.sklearn.realmlp import RealMLP_Tabular_Classifier
CatBoost_D_Regressor
from pytabkit import CatBoost_D_Regressor
LGBM_D_Classifier
from pytabkit import LGBM_D_Classifier

Quick example: train a RealMLP classifier on synthetic data with a time limit.

from pytabkit.models.sklearn.realmlp import RealMLP_Tabular_Classifier from sklearn.datasets import make_classification from sklearn.model_selection import train_test_split X, y = make_classification(n_samples=100, n_features=20, random_state=0) X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0) model = RealMLP_Tabular_Classifier( time_limit_s=30, # limit training time n_refit=1, device='cpu' ) model.fit(X_train, y_train) y_pred = model.predict(X_test) print(f'Accuracy: {sum(y_pred == y_test) / len(y_test):.3f}')
Debug
Known issues
breakingIn v1.5.0, RealMLP_Ensemble_Classifier and RealMLP_Ensemble_Regressor were removed. Use RealMLP_Tabular_Classifier with use_caruana_ensembling=True instead.
fix
Replace 'from pytabkit.models.sklearn.realmlp import RealMLP_Ensemble_Classifier' with 'from pytabkit.models.sklearn.realmlp import RealMLP_Tabular_Classifier' and add 'use_caruana_ensembling=True' to the constructor.
affects: >=1.5.0
breakingIn v1.6.1, default classification ensembling changed from averaging logits to averaging probabilities. This may change results.
fix
To revert to logit averaging, set 'ens_av_before_softmax=True' in the model constructor.
affects: >=1.6.1
gotchaWhen using extra models (GBDT, CatBoost, TabM), ensure you install pytabkit with the 'extra' option: pip install pytabkit[extra]. The base install only includes RealMLP.
fix
Install the extra dependencies: 'pip install pytabkit[extra]'.
affects: >=1.4.0
Upgrade
Version history
1.7.3latest on PyPI · released Jan 6, 2026
Audit
Dependencies
torchrequiredCore dependency for all neural network models.
scikit-learnrequiredUsed for compatibility (>=1.8).
pandasrequiredData handling.
numpyrequiredNumerical operations.
Agent activity
7 hits · last 30 days
node
6
OpenAI (training)
1
Resources