How to run forecasting task.

AI4HealthUOL / SSSD-ECG

Repository for the paper: 'Diffusion-based Conditional ECG Generation with Structured State Space Models'

MIT License

42 stars 6 forks source link

How to run forecasting task. #11

Closed vermabasu closed 8 months ago

vermabasu commented 8 months ago

Hi, I'm trying to run the forecasting task on PTB-XL (248) dataset. Can you clarify what all changes need to do to run forecasting task? I have tried all the previous solution mentioned in closed issues but getting errors. Also, while inference, what is the purpose of "number of utterance to be generated" argument?

juanlopezcode commented 8 months ago

Hi, I believe it is about SSSD model itself rather than this ecg application. The forecasting task is a blackout scenario, using mask iteratively sampled at different time steps windows per batch during training, and the --num-samples argument is about how many batch sizes while generating/imputing.