📚 TabSTAR: A Tabular Foundation Model for Tabular Data with Text Fields
🚀 TabSTAR v1.1 is on its way
We are releasing TabSTAR v1.1, a new generation of pretrained TabSTAR models. It ships alongside the original models, in its own module, so existing code keeps working unchanged:
| import | estimators | |
|---|---|---|
| TabSTAR | tabstar.tabstar_model |
TabSTARClassifier, TabSTARRegressor |
| TabSTAR v1.1 | tabstar.tabstarv1p1_model |
TabSTARv1p1Classifier, TabSTARv1p1Regressor |
We are releasing the model first. A detailed writeup of what changed, together with the v1.1 pretraining code, will follow.
Install
To fit a pretrained TabSTAR model to your own dataset, install the package using:
pip install tabstar
Quickstart Example
You can quickly get started with TabSTAR using the following example.
from importlib.resources import files
import pandas as pd
from sklearn.model_selection import train_test_split
from tabstar.tabstar_model import TabSTARClassifier
csv_path = files("tabstar").joinpath("resources", "imdb.csv")
x = pd.read_csv(csv_path)
y = x.pop('Genre_is_Drama')
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.1)
# For regression tasks, replace `TabSTARClassifier` with `TabSTARRegressor`.
tabstar = TabSTARClassifier()
tabstar.fit(x_train, y_train)
# tabstar.save("my_model_path.pkl")
# tabstar = TabSTARClassifier.load("my_model_path.pkl")
# y_pred = tabstar.predict(x_test)
metric = tabstar.score(X=x_test, y=y_test)
print(f"AUC: {metric:.4f}")
To use TabSTAR v1.1 instead, swap the import and the class name:
from tabstar.tabstarv1p1_model import TabSTARv1p1Classifier, TabSTARv1p1Regressor
tabstar = TabSTARv1p1Classifier()
tabstar.fit(x_train, y_train)
y_pred = tabstar.predict(x_test)
metric = tabstar.score(X=x_test, y=y_test)
fit, predict, predict_proba, score, save and load behave as they do for the original models.
Model cards: classification, regression.
For paper replication, TabSTAR evaluation on benchmarks, custom pretraining or research purposes, see:
🔗 TabSTAR Research Repository
Citation
If you use TabSTAR in your work, please cite:
@inproceedings{
arazi2025tabstar,
title={Tab{STAR}: A Tabular Foundation Model for Tabular Data with Text Fields},
author={Alan Arazi and Eilam Shapira and Roi Reichart},
booktitle={The Thirty-ninth Annual Conference on Neural Information Processing Systems},
year={2025},
url={https://openreview.net/forum?id=FrXHdcTEzE}
}
License
MIT © Alan Arazi et al.
1 maintainer
Metadata
1 maintainer
Release files for tabstar 1.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| tabstar-1.2.0.tar.gz | 184.2 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tabstar-1.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 382.4 kB
Release files / tabstar-1.2.0.tar.gz
| Download URL | tabstar-1.2.0.tar.gz |
|---|---|
| Size | 184.2 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
f05bd13e6d072647dd3af60b38412857e5aad89b88db9fc1d12e162685fc40e8
|
|
BLAKE2b-256 checksum How to use checksums |
cf4c22adb44a2b1c9dca1e3ae4003133462d4dd930908e069988d513e47cb953
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.0
|
Release files / tabstar-1.2.0-py3-none-any.whl
| Download URL | tabstar-1.2.0-py3-none-any.whl |
|---|---|
| Size | 198.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
9b82b7ec86ced256ef36bf680bda1562268ee5d926c6e1c48aa7bdc8bc8ea9f1
|
|
BLAKE2b-256 checksum How to use checksums |
c5fdc6608c2bd47b4e4072126642af27d40b79928bf73b3947aa5a7c01bb3db6
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.0
|
