Reinforcement learning environment for cognitive tests
CogEnv is a virtual environment based on AndroidEnv that allows artificial agents to complete the same tests used in human studies. It enables typical reinforcement learning agents to perform tasks commonly used in cognitive psychology, such as the N-Back and Stroop tests. CogEnv is designed to be a flexible and extensible platform for studying a wide range of cognitive tests and comparing the performance of artificial agents to human participants.