lucidrains/tab-transformer-pytorch
A PyTorch implementation of TabTransformer, a transformer-based model for tabular data classification and regression.

Not currently ranked — collecting fresh signals.
star history
This repository implements the TabTransformer architecture for tabular data, leveraging self-attention mechanisms from transformers. The model handles both categorical and continuous features, using depth-wise attention layers followed by MLP layers for prediction. It also includes the FT-Transformer variant, which applies transformer architecture more broadly to tabular data problems.
Frequently asked
- What is lucidrains/tab-transformer-pytorch?
- A PyTorch implementation of TabTransformer, a transformer-based model for tabular data classification and regression.
- Is tab-transformer-pytorch open source?
- Yes — lucidrains/tab-transformer-pytorch is open source, released under the MIT license.
- What language is tab-transformer-pytorch written in?
- lucidrains/tab-transformer-pytorch is primarily written in Python.
- How popular is tab-transformer-pytorch?
- lucidrains/tab-transformer-pytorch has 1.1k stars on GitHub.
- Where can I find tab-transformer-pytorch?
- lucidrains/tab-transformer-pytorch is on GitHub at https://github.com/lucidrains/tab-transformer-pytorch.