Model Fitting Code

This repository includes the code used for model fitting in de Jong et al. 2024

Figure 3

Rescorla-Wagner.m - MATLAB code used for fitting a model to data obtained during a two-armed bandit task.

Figure 4

two_stage_model.ipynb - Jupyter Notebook detailing the model fitting to our Neuropixels dataset.

For code used for the processing of raw data please see

• FIPSTER - Used for fiber photometry
• PKS - Used for in-vivo electrophysiology