← 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
Velocity · 7d
+0.5
★ / day
Trend
steady
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.

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