zenggo / cog_ml_tasks

This repo includes several tasks/environments built on openai-gym for training RL agents.

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

1. AX_12
2. AX_CPT