This is a simple OpenAI Gym-compatible environment to test RL agent's memory.
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"
)
- easy environment (partial observability maze with 4 different colors) :
"SimpleMemoryTestingEnv-Easy-v0"
(with only 2 colors:"SimpleMemoryTestingEnv-Easy-2Colors-v0"
)
This package is available in PyPi as simple_memory_testing_env
pip install simple_memory_testing_env
git clone https://www.github.com/Near32/SimpleMemoryTestingEnv
cd SimpleMemoryTestingEnv
pip install -e .