Update README.md

Author: Jathurshan0330

README.md

@@ -38,6 +38,11 @@ pip install -r requirements.txt
 
 Kindly refer to [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1hs6V6doKpqlLlPOsnKllZp9ATuA9VqA7?usp=sharing) to extract dataset to train the models.
 
+```
+python "./data_preparations/single_epoch.py" --save_path  "path/to/save/dataset"
+```
+
+
 ## Inference and Get Interpretations
 
 Kindly refer to [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1cBDzzzwdyBp-bpxR_wvVzVlxrbSwrbiG?usp=sharing) to get prediction for any subject in sleepedf dataset and get interpretation results as given in the paper.