Fitting a simple Reinforcement Learning model to behavioral data with PyMC3 (Jupyter NB)

Hello,

I recently wrote a Jupyter Notebook illustrating how one can fit a very simple RL model to simulated behavioral data. I thought this could be useful for others, specially when writing models that require recursive calculations based on sampled parameters:

Link: https://github.com/ricardoV94/stats/blob/master/modelling/RL_PyMC.ipynb

Have a nice day :slight_smile:

6 Likes

Thank you for sharing, this looks great :tada:

2 Likes

FYI: @Maria also share her RL models on human data here Modeling reinforcement learning of human participant using PyMC3

4 Likes

Iā€™m loving this sharing mood :wink:

2 Likes