# Running Imputation Experiments
## Download datasets

Download the zip folders namely : ETT-small.zip (for ETTh1, ETTh2, ETTm1 and ETTm2), weather.zip and electricity.zip from https://drive.google.com/drive/folders/1vE0ONyqPlym2JaaAoEe0XNDR8FS_d322  
Creat a folder named 'datasets' in the tspulse/imputation folder. 
Place the extracted folders in the 'datasets' folder. 

The datasets folder structure with the csv files should be like:
<pre> 
datasets
├── electricity
│   └── electricity.csv
├── ETT-small
│   ├── ETTh1.csv
│   ├── ETTh2.csv
│   ├── ETTm1.csv
│   └── ETTm2.csv
└── weather
    └── weather.csv
</pre>

## Zero-Shot Experiments:
Run the notebook imputation_zeroshot.ipynb and provide the dataset name (dset), mask ratios(m_r) and mask type(m_t) values in cell 5. 

## Finetuning Experiments:
Run the notebook imputation_finetune.ipynb and provide the DATASET, mask_ratio and mask_type values in cell 6. 