vwxyzjn / entity-gym

Standard interface for entity based reinforcement learning environments.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

License:Other


Languages

Language:Python 100.0%