Code for the paper - 'Modeling the Role of the Striatum in Non-Stationary Bandit Tasks' - View it on GitHub