fitr 0.0.1

Fit reinforcement learning models to behavioural data

Python implementation of package to fit reinforcement learning models to behavioural data

Installation

The current PyPI release of Fitr can be installed as follows:

pip install fitr

If you want the latest version on the GitHub master branch, install as follows:

pip install git+https://github.com/ComputationalPsychiatry/fitr.git

Tutorials

Tutorials (Jupyter Notebooks) can be found in the examples folder. They include

1. Introductory tutorial (EM and Bayesian Model Selection)

2. Fitting a Model with MCMC

3. Use MCMC with your own Stan Code

4. Using Multiple Model-Fitting Routines for Same Model

How to Cite

If you use Fitr in your work, we would very much appreciate the citation, which can be done as follows:

• Abraham Nunes, Alexander Rudiuk, & Thomas Trappenberg. (2017). Fitr: A Toolbox for Computational Psychiatry Research. Zenodo. http://doi.org/10.5281/zenodo.439989