/TDSTF

Primary LanguagePythonMIT LicenseMIT

TDSTF

This is the github repository for the paper "TDSTF: Transformer-based Diffusion probabilistic model for Sparse Time series Forecasting"

MIMIC-III data

Download the dataset at https://physionet.org/content/mimiciii-demo/1.4/

Data preprocessing

Run the files step_1.py through step_4.py in order in the folder "preprocess"

Experiments

Run the file main.py

To test a pretrained model, please assign the model folder name to the parameter "modelfolder"