AI4HealthUOL / SSSD

Repository for the paper: 'Diffusion-based Time Series Imputation and Forecasting with Structured State Space Models'
MIT License
270 stars 47 forks source link

on ecg_utils.py #22

Closed chomgid closed 6 months ago

chomgid commented 6 months ago

I'm currently trying to reproduce experiment results from the "Diffusion-based Time Series Imputation and Forecasting with Structured State Space Models" paper, especially on PTB-XL dataset.

From your code in SSSD/docs/instructions/PTB-XL/clinical_ts/ecg_utils.py, line 25 'from .stratify import stratify,stratify_batched' causes error in my machine saying "ImportError: attempted relative import with no known parent package" This is not in any of the other folders, nor it is a python package. Can you please check on this? And also 'get_stratified_kfolds' function from the same file is not defined.

Thank you!

juanlopezcode commented 6 months ago

I have uploaded the file