--- id: pytabkit version: "1.7.3" license: Apache-2.0 license_treatment: permissive maintenance: aging --- # pytabkit — ML models + benchmark for tabular data classification and regression License: permissive · Maintenance: aging · Downloads: 102.5K/mo ## What it is and what it does PyTabKit wraps modern tabular machine learning methods—neural networks like RealMLP and TabM, gradient boosted trees, and hybrid models—behind scikit-learn-compatible interfaces. It automatically handles GPU selection, categorical column detection, numerical preprocessing, and train-validation splitting for early stopping, with optional cross-validation ensembling and hyperparameter optimization. The package is built around a NeurIPS 2024 paper benchmarking these methods on tabular classification and regression tasks. It includes both the trained model interfaces and the benchmarking code used to evaluate them. While the documentation recommends AutoGluon for best-possible results in production, PyTabKit is positioned as easier to use for development and supports vectorized cross-validation that can accelerate training of certain models. Use it for: - Quick prototyping of tabular classifiers and regressors with automatic preprocessing and GPU acceleration. - Benchmarking multiple tabular ML methods on your own datasets using the built-in benchmarking utilities. - Training RealMLP or TabM models with cross-validation ensembling and hyperparameter optimization for improved accuracy. - Comparing gradient boosted tree implementations (XGBoost, LightGBM, CatBoost) with neural network baselines on structured data. - Post-hoc calibration and refinement stopping for probabilistic predictions using temperature scaling. ## Worth the install? AI-flagged interpretation of the facts above — verify before relying. PyTabKit provides scikit-learn interfaces to modern tabular machine learning models—including RealMLP, gradient boosted trees (XGB, LGBM, CatBoost), and TabR—with built-in hyperparameter optimization, cross-validation ensembling, and benchmarking utilities. Yes, with conditions. Install if you are developing tabular ML pipelines and want scikit-learn-compatible interfaces to modern methods with minimal setup friction. The low install overhead and automatic preprocessing are valuable for iteration. However, if your goal is production-grade best performance, the documentation explicitly recommends AutoGluon instead. Maintenance is aging but not stale; no known vulnerabilities. ## Install pip install pytabkit uv add pytabkit poetry add pytabkit ## Installing pytabkit Before you install: Low install friction; pure Python wheel. Requires torch, pytorch-lightning, and scikit-learn as core runtime dependencies. Status is aging (220 days since last release), though the repository remains active and well-maintained with recent commits and no archived status. License in practice: Apache-2.0 permissive license allows commercial and private use with minimal restrictions; suitable for most production and research contexts. Quickstart: pip install pytabkit from pytabkit import RealMLP_TD_Classifier model = RealMLP_TD_Classifier() model.fit(X_train, y_train) model.predict(X_test) Missing numerical values are not allowed and must be imputed beforehand. Requires torch to be installed separately if you need to control CPU/GPU version. TabR requires manual faiss installation (conda only). Verify before relying: - Whether vectorized cross-validation speedup for RealMLP is significant enough to justify adoption over AutoGluon for production use cases. - Current performance comparison with TabArena's newer models and preprocessing capabilities for real-world datasets. - GPU memory requirements and scalability limits for large tabular datasets. ## Package facts - License: Apache-2.0 (permissive) - Python support: supports_current - Install friction: low - Maintenance: aging - Downloads: 102.5K/month (top 15,000 on PyPI) - Known vulnerabilities: none known ## Tags tabular machine learning models, scikit-learn tabular classifiers, neural networks for tabular data, gradient boosting wrappers, tabular data benchmarking, AutoML for structured data, RealMLP classifier, ensemble tabular models, tabular-ml, benchmarking, neural-networks [View on SkillFed](https://skillfed.io/packages/pytabkit) · [View on PyPI](https://pypi.org/project/pytabkit/)