/two-step-task

Implementation of the two-step-task as described in "Prefrontal cortex as a meta-reinforcement learning system" and "Learning to Reinforcement Learn".

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers