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

which leads each index from 1 to 12 corresponds to? #16

Closed Ssw2001 closed 4 months ago

Ssw2001 commented 4 months ago

The shape of the dataset you provided, test_ptbxl_1000.npy, is (2203, 12, 1000). I would like to inquire about which leads each index from 1 to 12 corresponds to? thank you!

hayaalsh commented 4 months ago

The PTB-XL dataset uses the following lead order: ['I', 'II', 'III', 'AVR', 'AVL', 'AVF', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6']. Note that the order listed on the original dataset website is incorrect; they switch 'AVR' and 'AVL'.

juanlopezcode commented 4 months ago

Hi, if you are using the ecg_data_preprocessing.ipynb file to extract the data, then you get the order of ['I','II','V1',...,'V6','III','AVR','AVL','AVF']. Otherwise, if you processed the data directly from physionet, you get the standard ['I', 'II', 'III', 'AVR', 'AVL', 'AVF', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6'] mentioned in the previous comment, I hope this helps Best Juan

hayaalsh commented 4 months ago

How about the synthetic data generated by the model? Does it follow ['I','II','V1',...,'V6','III','AVR','AVL','AVF'] as well?

Update: just checked and the synthetic data has same order you listed. Thank you :)

Ssw2001 commented 4 months ago

Hi, if you are using the ecg_data_preprocessing.ipynb file to extract the data, then you get the order of ['I','II','V1',...,'V6','III','AVR','AVL','AVF']. Otherwise, if you processed the data directly from physionet, you get the standard ['I', 'II', 'III', 'AVR', 'AVL', 'AVF', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6'] mentioned in the previous comment, I hope this helps Best Juan

Thank you for your reply ! i directly used the test_ptbxl_1000.npy and the train_ptbxl_1000.npy from https://mega.nz/folder/kT91jYpI#97GyTkVVUk97fzs1Oy4nBQ/folder/sW8UiBpT you provide as my data, do the test_ptbxl_1000.npy and the train_ptbxl_1000.npy both follow the order ['I', 'II', 'III', 'AVR', 'AVL', 'AVF', 'V1', 'V2', 'V3', 'V4', 'V5', 'V6'] or ['I','II','V1',...,'V6','III','AVR','AVL','AVF']? this is very important to me,thank you!

juanlopezcode commented 4 months ago

['I','II','V1',...,'V6','III','AVR','AVL','AVF']