← all repositories

lucidrains/tab-transformer-pytorch

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

1.1k stars Python ML Frameworks
tab-transformer-pytorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.