chenghuzi / pyrl

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Reward-based training of RNNs

INFO

Forked from song's repo

  • Add T.Yang's Task paradigm (2007) to this repo.
  • Transplant some of the code (task and training) to some other project in neuroscience.

Requirements

This code is written in Python 2.7 and requires

Notes

  • For the paper, we used a time step of 10 ms, which results in hundreds of time steps for typical tasks. When training a new task, we highly recommend that you start with a larger value to save time.

License

MIT

Citation

  • Song, H. F., Yang, G. R., & Wang, X.-J. (2017) Reward-based training of recurrent neural networks for cognitive and value-based tasks. eLife, in press.

About

License:MIT License


Languages

Language:Python 100.0%