https://github.com/priorlabs/tabpfn-extensions

Community extensions for TabPFN - the foundation model for tabular data. Built with TabPFN! 🤗
https://github.com/priorlabs/tabpfn-extensions

data-science machine-learning tabpfn tabular-data

Last synced: 11 months ago
JSON representation

Community extensions for TabPFN - the foundation model for tabular data. Built with TabPFN! 🤗

• Host: GitHub
• URL: https://github.com/priorlabs/tabpfn-extensions
• Owner: PriorLabs
• License: apache-2.0
• Created: 2024-12-05T10:43:40.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2025-03-28T20:25:42.000Z (11 months ago)
• Last Synced: 2025-03-29T20:02:26.896Z (11 months ago)
• Topics: data-science, machine-learning, tabpfn, tabular-data
• Language: Python
• Homepage: https://www.priorlabs.ai
• Size: 650 KB
• Stars: 90
• Watchers: 3
• Forks: 16
• Open Issues: 13
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# TabPFN Extensions ⚡

[![PyPI version](https://badge.fury.io/py/tabpfn-extensions.svg)](https://badge.fury.io/py/tabpfn-extensions)

[![Downloads](https://pepy.tech/badge/tabpfn)](https://pepy.tech/project/tabpfn)

[![License](https://img.shields.io/badge/License-Apache_2.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)

[![Discord](https://img.shields.io/discord/1285598202732482621?color=7289da&label=Discord&logo=discord&logoColor=ffffff)](https://discord.com/channels/1285598202732482621/)

[![Twitter Follow](https://img.shields.io/twitter/follow/Prior_Labs?style=social)](https://twitter.com/Prior_Labs)

![Contributions Welcome](https://img.shields.io/badge/contributions-welcome-brightgreen)

![Last Commit](https://img.shields.io/github/last-commit/automl/tabpfn-client)



## 🛠️ Available Extensions

- **post_hoc_ensembles**: Improve performance with model combination

- **interpretability**: Explain TabPFN predictions with SHAP values and feature selection

- **many_class**: Handle classification with more classes than TabPFN's default limit

- **classifier_as_regressor**: Use TabPFN's classifier for regression tasks

- **hpo**: Automatic hyperparameter tuning for TabPFN

- **rf_pfn**: Combine TabPFN with decision trees and random forests

- **unsupervised**: Data generation and outlier detection

- **embedding**: Get TabPFNs internal dense sample embeddings

Detailed documentation for each extension is available in the respective module directories.

## ⚙️ Installation

```bash

# Clone and install the repository

pip install "tabpfn-extensions[all] @ git+https://github.com/PriorLabs/tabpfn-extensions.git"

```

### 🔄 Backend Options

TabPFN Extensions works with two TabPFN implementations:

1. **🖥️ TabPFN Package** - Full PyTorch implementation for local inference:

   ```bash

   pip install tabpfn

   ```

2. **☁️ TabPFN Client** - Lightweight API client for cloud-based inference:

   ```bash

   pip install tabpfn-client

   ```

Choose the backend that fits your needs - most extensions work with either option!

## 📝 License

This project is licensed under the Apache License 2.0 - see the [LICENSE](LICENSE) file for details.

## 🧑‍💻 For Contributors

Interested in adding your own extension? We welcome contributions!

```bash

# Clone and set up for development

git clone https://github.com/PriorLabs/tabpfn-extensions.git

cd tabpfn-extensions

# Lightweight dev setup (fast)

pip install -e ".[dev]"

# Test your extension with fast mode

FAST_TEST_MODE=1 pytest tests/test_your_extension.py -v

```

See our [Contribution Guide](CONTRIBUTING.md) for more details.

[![Contributors](https://contrib.rocks/image?repo=priorlabs/tabpfn-extensions)](https://github.com/priorlabs/tabpfn-extensions/graphs/contributors)

## 📦 Repository Structure

Each extension lives in its own subpackage:

```

tabpfn-extensions/

├── src/

│   └── tabpfn_extensions/

│       └── your_package/      # Extension implementation

├── examples/

│   └── your_package/          # Usage examples

└── tests/

    └── your_package/          # Tests

```

---

Built with ❤️ by the TabPFN community