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