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

 
 
 

Contributors