Metadata-Version: 2.4
Name: tabimpute
Version: 0.0.4
Summary: TabImpute: A Pre-trained Transformer for Missing Data Imputation
Author: Jacob Feitelberg, Dwaipayan Saha, Zaid Ahmad, Kyuseong Choi, Anish Agarwal, Raaz Dwivedi
Keywords: foundation model,in-context learning,missing data imputation,tabular data
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering
Requires-Python: <3.13,>=3.9
Requires-Dist: einops>=0.7
Requires-Dist: huggingface-hub
Requires-Dist: networkx
Requires-Dist: numpy
Requires-Dist: tabpfn==6.0.0
Requires-Dist: torch<3,>=2.2
Provides-Extra: benchmark
Requires-Dist: joblib; extra == 'benchmark'
Requires-Dist: matplotlib; extra == 'benchmark'
Requires-Dist: pandas; extra == 'benchmark'
Requires-Dist: psutil; extra == 'benchmark'
Requires-Dist: scikit-learn==1.4.2; extra == 'benchmark'
Requires-Dist: scipy; extra == 'benchmark'
Requires-Dist: tabpfn; extra == 'benchmark'
Requires-Dist: tqdm>=4.64.0; extra == 'benchmark'
Requires-Dist: transformers; extra == 'benchmark'
Requires-Dist: wandb; extra == 'benchmark'
Requires-Dist: xgboost; extra == 'benchmark'
Provides-Extra: categorical
Requires-Dist: scipy; extra == 'categorical'
Provides-Extra: preprocessing
Requires-Dist: scipy; extra == 'preprocessing'
Provides-Extra: tabpfn-extensions
Requires-Dist: tabpfn-extensions; extra == 'tabpfn-extensions'
Provides-Extra: training
Requires-Dist: joblib; extra == 'training'
Requires-Dist: pandas; extra == 'training'
Requires-Dist: psutil; extra == 'training'
Requires-Dist: scikit-learn==1.4.2; extra == 'training'
Requires-Dist: scipy; extra == 'training'
Requires-Dist: tqdm>=4.64.0; extra == 'training'
Requires-Dist: transformers; extra == 'training'
Requires-Dist: wandb; extra == 'training'
Requires-Dist: xgboost; extra == 'training'
Description-Content-Type: text/markdown

# TabImpute

TabImpute is a pre-trained transformer for missing data imputation on tabular data.

This code is based on the TabPFN and TabICL codebases, both available on GitHub.