Code to fit Bayesian versions of simple reinforcement learning models