Part1: Tensorflow version of SSSD of Paper Diffusion-based Time Series Imputation and Forecasting with Structured State Space Models, the official GitHub page is:https://github.com/AI4HealthUOL/SSSD.
Part2: Tensorflow version of Mega of Paper Mega: Moving Average Equipped Gated Attention, the official GitHub page is:https://github.com/facebookresearch/mega
-
run train_imputers.py to train SSSD/CSDI/Mega for imputations.
-
run evaluate_sssd/csdi/mega.py to genegrate and evaluate the missing data results.
-
run predict.py to do the predcition