This repository contains code and experiments for evaluating the BrainUnit framework. The experiments focus on training and visualizing neural network models.
To install the required dependencies, use pip:
pip install -r requirements.txt├── 01-hh-neurons/
├── 02-mutiscale-network/
│ └── results/
├── 03-hh-fitting/
├── 04-task-training/
│ ├── results/
│ ├── task_training.py
│ └── verification.py
├── archive/
├── .gitignore
├── README.md
└── requirements.txt
01-hh-neurons/: Contains experiments related to simulating Hodgkin-Huxley-styled TRN neurons.02-mutiscale-network/: Contains results related to multiscale network experiments.03-hh-fitting/: Contains experiments related to fitting Hodgkin-Huxley models.04-task-training/: Contains the main training script and results for training evidence accumulation tasks..gitignore: Specifies files and directories to be ignored by Git.README.md: This file.requirements.txt: Lists the dependencies required for the project.
For the Hodgkin-Huxley neuron experiments, defined in 01-hh-neurons/ directory, to run the script, execute:
python dendritex-sim.pyFor the multiscale spiking network experiments, defined in 02-multiscale-network/ directory, to run the script, execute:
python large_scale_COBA_EI-bst.py # with physical units
python large_scale_COBA_EI-bp.py # without physical unitsFor the Hodgkin-Huxley model fitting experiments, defined in 03-hh-fitting/ directory, to run the script, execute:
python brian2_hh_fitting.py # fitting with brian2
python neuron_fitting_of_hh_model.py # fitting with dendritexFor the cognitive task training task, defined in 04-task-training/ directory, to run the training script, execute:
python task_training.pyContributions are welcome! Please open an issue or submit a pull request for any improvements or bug fixes.
If you use this code in your research, please consider citing the following paper: