← all repositories

dreamquark-ai/tabnet

A PyTorch library implementing TabNet, an attention-based deep learning model designed for tabular data classification and regression tasks.

2.9k stars Python ML Frameworks
tabnet
Velocity · 7d
+1.2
★ / day
Trend
steady
star history

TabNet provides a PyTorch implementation of the TabNet architecture, which uses sequential attention to choose which features to reason from at each decision step. The model combines the interpretability of decision trees with the representational power of deep neural networks, enabling both high accuracy on tabular data and feature importance insights through its attention mechanism. It is distributed via pip and conda as pytorch-tabnet.

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