Official PyTorch implementation of DreaMR described in the paper.
- python >= 3.7
- numpy >= 1.21.4
- torch >= 1.10.1
- torch-cuda >= 10.2
- torchvision >= 0.11.2
- tqdm >= 4.62.3
We use three datasets for our experiments in the paper, HCP-Rest, HCP-Task, ID1000 datasets. Due to privacy concerns, we are unable to share the datasets. But you can download them from their official sites.
But we provide a dummy data loader to show what is expected by the dataset.py file.
Here we provide an example transformer based classifier (official repo here) which is used for the counterfactual generations.
python main.py --targetDataset datasetName --method bolT_classify --do train
Please note that you have to implement the dataset loader and do the required changes in the code to import your loader. Also based on the dataset your using, set the correct variables (nOfClasses, dynamicLength) in the main file.
