Awesome
Cognitive and ML Tasks
This repo includes several tasks/environments built on openai-gym for training RL agents.
The original paper: Martinolli, Marco, Wulfram Gerstner, and Aditya Gilra. ‘Multi-Timescale Memory Dynamics Extend Task Repertoire in a Reinforcement Learning Network With Attention-Gated Memory’. Frontiers in Computational Neuroscience 12 (2018).
Install
$ git clone https://github.com/CgnRLAgent/cog_ml_tasks.git
$ cd cog_ml_tasks
$ pip install -e .
Requirements: see the setup.py.
Envs/Tasks Descriptions
- 12_AX
- AX_CPT
- 12_AX_S
- 12_AX_CPT
- Simple_Copy
- Simple_Copy_Repeat
- Sequence Prediction
- Saccade