/tfSSSD

Tensorflow version of SSSD

Primary LanguagePython

tfSSSD

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.

tfMega

Part2: Tensorflow version of Mega of Paper Mega: Moving Average Equipped Gated Attention, the official GitHub page is:https://github.com/facebookresearch/mega

How to run codes

  1. run train_imputers.py to train SSSD/CSDI/Mega for imputations.

  2. run evaluate_sssd/csdi/mega.py to genegrate and evaluate the missing data results.

  3. run predict.py to do the predcition