← all repositories

WenjieDu/SAITS

A PyTorch-based deep learning model that uses self-attention to impute missing values in multivariate time series.

505 stars Python ML FrameworksDomain Apps
SAITS
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

SAITS is a PyTorch implementation of a deep learning model for time-series imputation, using self-attention mechanisms to reconstruct incomplete multivariate time series containing NaN values. The model applies dual self-attention blocks to jointly capture temporal dependencies and feature correlations for imputation tasks. The repository includes training utilities, benchmark scripts, and integrations with other time-series models via the PyPOTS framework.

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