search

bakqui / ST-MEM

[ICLR2024] Guiding Masked Representation Learning to Capture Spatio-Temporal Relationship of Electrocardiogram
Other
16 stars 0 forks source link

Pretrained model selection #5

Open sehunfromdaegu opened 4 days ago

sehunfromdaegu commented 4 days ago

I'm excited to run the code, thanks for uploading.

  1. I wonder how the pretrained model was selected for downstream task. Did you use the the pretrained model after 800 epochs? Or, pretrained model was selected among all epochs that work best for each downstream tasks?

  2. Also, pretrained dataset code15 is not prepared in WFDB format, so I can't run /data/process_data.ecg.

  3. How can I prepare a dataset for downstream task?