Reinforcement learning + drift-diffusion modeling - View it on GitHub