Contextual modulation of value signals in reward and punishment learning

Please find attached three script that can be used to fit three models and the data from the paper:
1) A Basic Q-learning (not presented in the Paper)2) the Q-learning + counterfactual learning (the ABSOLUTE model in the paper)3) the RELATIVE model.
The Computational_Model_* file is the one actually containing the models' algorithms. The other codes serve to fit the models to the data with both classical (LL) and "bayesian" (LPP) techniques. I commented almost all the lines to make the code as much as accessible as I can. However if anything is not clear please do not hesitate to contact me, I will be very happy to help. The code there is written for session 1, but it can be easily adapted for other sessions. I also join the date from the NAture communication study (28 subjects, 4 session each).