Add pipeline tag

README.md

@@ -1,20 +1,20 @@
 ---
 license: mit
+pipeline_tag: feature-extraction
 ---
+
 # ECGFounder: An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains
 
 This is the official implementation of our paper "[An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains](https://arxiv.org/abs/2410.04133)".
 
 > Authors: Jun Li, Aaron Aguirre, Junior Moura, Jiarui Jin, Che Liu, Lanhai Zhong, Chenxi Sun, Gari Clifford, Brandon Westover, Shenda Hong.
 
-
+For the code, see the [GitHub repository](https://github.com/PKUDigitalHealth/ECGFounder.git).
 
 ## 🚀 Getting Started
 
 🚩 **News** (Mar 2025): The pre-training checkpoint is now available on [🤗 Hugging Face](https://huggingface.co/PKUDigitalHealth/ECGFounder/tree/main)!
 
-
-
 ### Installation
 
 To clone this repository:
@@ -33,22 +33,16 @@ conda activate ECGFounder
 pip install -r requirements.txt
 ```
 
-
-
 ### Fine-tune on Downstream Tasks
 
 In our paper, downstream datasets we used are as follows:
 
-* **MIMIC-ECG**: Please download the [MIMIC-ECG](https://physionet.org/content/mimiciv/2.2/) dataset from physionet.
-
-Next, please download the model's checkpoint from the  [🤗 Hugging Face](https://huggingface.co/PKUDigitalHealth/ECGFounder/tree/main). And place the model weights in path *./checkpoint*
-
+*   **MIMIC-ECG**: Please download the [MIMIC-ECG](https://physionet.org/content/mimiciv/2.2/) dataset from physionet.
 
+Next, please download the model's checkpoint from the [🤗 Hugging Face](https://huggingface.co/PKUDigitalHealth/ECGFounder/tree/main). And place the model weights in path *./checkpoint*
 
 You can run the jupyter notebook to finetune the model by the example dataset.
 
-
-
 ## References
 
 If you found our work useful in your research, please consider citing our works at:
@@ -56,7 +50,7 @@ If you found our work useful in your research, please consider citing our works
 > ```
 > @article{li2024electrocardiogram,
 > title={An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains},
-> author={Li, Jun and Aguirre, Aaron and Moura, Junior and Liu, Che and Zhong, Lanhai and Sun, Chenxi and Clifford, Gari and Westover, Brandon and Hong, Shenda},
+> author={Li, Jun and Aguirre, Aaron and Moura, Junior and Liu, Che and Zhong, Lanhai and Sun, Chenxi and Gari Clifford and Brandon Westover, Shenda Hong},
 > journal={arXiv preprint arXiv:2410.04133},
 > year={2024}
 > }