Skip to content

Tutorial codes for modeling brains with neural nets

License

Notifications You must be signed in to change notification settings

gyyang/nn-brain

Repository files navigation

nn-brain

Status: In Maintenance.

Code for Artificial neural networks for neuroscientists: A Primer

Arxiv version

This repository provides tutorial-style code for training artificial neural networks on simple neuroscience-relevant tasks, and for analyzing these networks using a variety of neuroscience methods.

Content

(1) Jupyter notebook for training convolutional neural networks, representation-similarity-analysis (RSA), and gradient-based tuning analysis.

(2) Jupyter notebook for training LSTMs on a memory task, and visualizing neural activity.

(3) Jupyter notebook for training RNNs on a working memory task, and fixed-point-based dynamical system analysis.

(4) Jupyter notebook for training Excitatory-Inhibitory RNNs on a decision-making task, and analyzing network connectivity.

Requirements

This code is tested with Python 3 and a recent version of Pytorch. Please kindly let me know if it does not work for you. All code can be run quickly (a few minutes) on a laptop CPU.

Please email me at gyyang.neuro@gmail.com or send issues/pull requests, if you have any feedback. Thank you!

About

Tutorial codes for modeling brains with neural nets

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published