📚 TabSTAR: A Tabular Foundation Model for Tabular Data with Text Fields
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}")
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.1.16
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.1.16.tar.gz | 150.4 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tabstar-1.1.16-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 301.4 kB
Release files / tabstar-1.1.16.tar.gz
| Download URL | tabstar-1.1.16.tar.gz |
|---|---|
| Size | 150.4 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
ff8bd7943bb5f379138a776371cb76245ea2aad93a522886dc5abe25e4c036cf
|
|
BLAKE2b-256 checksum How to use checksums |
4a588b0bf01352685fdcdda83528bc2bb07abe37f874c65699033ee8af658048
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.11.14
|
Release files / tabstar-1.1.16-py3-none-any.whl
| Download URL | tabstar-1.1.16-py3-none-any.whl |
|---|---|
| Size | 150.9 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
7be15f1f3501908ec4969da0f14d2ce62ba8eceeebbcf1bee6112cf444c3730e
|
|
BLAKE2b-256 checksum How to use checksums |
babdbe3b86e17f0a8f73a67d71f28dcc5ac11ae7f7d6834bb6af61e0c60b5d91
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.11.14
|
