-
Install dependencies
Run the following command in your terminal:pip install -r requirements.txt
If there is an error requiring the install of Rust and Cargo, please run the following command:
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
You might want to delete this from your PATH variables after running the code
-
Place the scan folders
- Move
Methodology1_Scansinto theMethodology #1directory. - Move
Methodology2_Scansinto theMethodology #2directory.
Both folders are available in the Canvas assignment titled βWeek 13: Final Report and Presentationβ Note:This project uses python 3.10
- Move
Note: If there is a file not found error, the "path" variable inside the codes can be changed to the proper folders necessary
-
Navigate to the
Methodology #1directory. -
Run the following scripts in order:
-
Note: You do not have to run these commands, they are here to let you know which files to run
python 'Methodology #1/Visiontransformer.py' python 'Methodology #1/XGBoost_ML.py'
- Navigate to the
Methodology #2directory. - Run the following scripts in sequence:
python 'Methodology #2/Create_FC_Dataset.py' python 'Methodology #2/FC_NN_Stage1.py' python 'Methodology #2/FC_NN_Stage2.py'
OCD_Classification_Project/
β
βββ Methodology #1/
β βββ Methodology1_Scans/
β βββ Visiontransformer.py
β βββ XGBoost_ML.py
β
βββ Methodology #2/
β βββ Methodology2_Scans/
β βββ Create_FC_Dataset.py
β βββ FC_NN_Stage1.py
β βββ FC_NN_Stage2.py
β
βββ requirements.txt
This project implements two distinct pipelines for classifying OCD patients based on neuroimaging data:
- Methodology #1: Image-based classification using Vision Transformers and XGBoost.
- Methodology #2: Functional connectivity matrix construction followed by a two-stage neural network classifier.
For questions or collaboration inquiries, please reach out via Canvas or email.