← all repositories

somepago/saint

A PyTorch implementation of SAINT, a deep learning model for tabular data using row attention and contrastive pre-training.

466 stars Python ML Frameworks
saint
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository provides the official implementation of the SAINT paper, introducing improved neural network architectures for tabular data. The model employs row-level attention mechanisms and contrastive pre-training strategies to enhance performance on tabular classification and regression tasks. Built in PyTorch, it supports various attention variants including column-only, row-only, and combined approaches, and can leverage unlabeled data for unsupervised pretraining to improve performance with limited labeled samples.

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