This code is used to reproduce the results of the CURBD repository (https://github.com/rajanlab/CURBD) associated with the paper Inferring brain-wide interactions using data-constrained recurrent neural network models from Matthew G. Perich and al. using the NeuroTorch library.
To run the code, you need to install the requirements in requirements.txt.
pip install -r requirements.txt
Then, you can run the code by running the following command:
python main.py
The repository is structured as follows:
main.py
: main file to run the codecurbd_dataset.py
: dataset class to construct the curbd synthetic datasetcurbd_training.py
: training functionfigures_script.py
: script with functions to generate the figuresutils.py
: utility functionsts_dataset.py
: dataset class to construct a time series dataset from a npy filemain_other_dataset.py
: main file to run the code on other datasets