Skip to content

This is a simple OpenAI Gym-compatible environment to test RL agent's memory.

License

Notifications You must be signed in to change notification settings

Near32/SimpleMemoryTestingEnv

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SimpleMemoryTestingEnv

This is a simple OpenAI Gym-compatible environment to test RL agent's memory.

Usage

gym must be installed. Environments can be created as follows:

>>> import gym
>>> import simple_memory_testing_env
>>> env = gym.make("SimpleMemoryTestingEnv-v0")
  • default environment (partial observability + memory-requiring T-maze with 4 different colors) : "SimpleMemoryTestingEnv-v0" (with only 2 colors: "SimpleMemoryTestingEnv-2Colors-v0")

default_env

  • easy environment (partial observability maze with 4 different colors) : "SimpleMemoryTestingEnv-Easy-v0" (with only 2 colors: "SimpleMemoryTestingEnv-Easy-2Colors-v0")

easy_env

Installation

Installing via pip

This package is available in PyPi as simple_memory_testing_env

pip install simple_memory_testing_env

Installing via cloning this repository

git clone https://www.github.com/Near32/SimpleMemoryTestingEnv
cd SimpleMemoryTestingEnv
pip install -e .