AI4HealthUOL / SSSD-ECG

Repository for the paper: 'Diffusion-based Conditional ECG Generation with Structured State Space Models'
MIT License
36 stars 4 forks source link

No module named 'clinical_ts.stratify' #3

Closed XLIU430 closed 1 year ago

XLIU430 commented 1 year ago

When I run the jupyter notebook 'ecg_data_preprocessing.ipynb'. There is an error in ptb_xl/clinical_ts/ecg_utils.py:31. Which is the 'from .stratify import stratify,stratify_batched'

juanlopezcode commented 1 year ago

Hello @XLIU430, you are right, there was missing a stratify.py file. I have added it. It should run now.