Skip to content

Standard interface for entity based reinforcement learning environments.

License

Notifications You must be signed in to change notification settings

entity-neural-network/entity-gym

Repository files navigation

Entity Gym

Actions Status PyPI Documentation Status Discord

Entity Gym is an open source Python library that defines an entity based API for reinforcement learning environments. Entity Gym extends the standard paradigm of fixed-size observation spaces by allowing observations to contain dynamically-sized lists of entities. This enables a seamless and highly efficient interface with simulators, games, and other complex environments whose state can be naturally expressed as a collection of entities.

The enn-trainer library can be used to train agents for Entity Gym environments.

Installation

pip install entity-gym

Usage

You can find tutorials, guides, and an API reference on the Entity Gym documentation website.

Examples

A number of simple example environments can be found in entity_gym/examples. More complex examples can be found in the ENN-Zoo project, which contains Entity Gym bindings for Procgen, Griddly, MicroRTS, VizDoom, and CodeCraft.

About

Standard interface for entity based reinforcement learning environments.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages