Welcome to aind-behavior-gym documentation!

Implementing AIND behavior tasks using gymnasium and setting up a base class for agents that perform the tasks.

What is this library for?

It aims to provide a common language among behavior-related animal training, artificial agent training, model fitting, and model simulation in AIND.

See related repositories:

Example Image

Structure

For now, three dynamic foraging tasks have been implemented.

  • Coupled block task

  • Uncoupled block task

  • Random walk task

classes_aind-behavior-gym

To develop more dynamic foraging tasks, please subclass DynamicForagingTaskBase.

Indices and tables