Updated on 2021.08.03 DI-engine-v0.1.1 (beta)
DI-engine is a generalized Decision Intelligence engine. It supports most basic deep reinforcement learning (DRL) algorithms, such as DQN, PPO, SAC, and domain-specific algorithms like QMIX in multi-agent RL, GAIL in inverse RL, and RND in exploration problems. Various training pipelines and customized decision AI applications are also supported. Have fun with exploration and exploitation.
- DI-engine-docs
- treevalue (preview)
You can simply install DI-engine from PyPI with the following command:
pip install DI-engineIf you use Anaconda or Miniconda, you can install DI-engine from conda-forge through the following command:
conda install -c opendilab di-engineFor more information about installation, you can refer to installation.
The detailed documentation are hosted on doc(中文文档).
Bonus: Train RL agent in one line code:
ding -m serial -e cartpole -p dqn -s 0We appreciate all contributions to improve DI-engine, both algorithms and system designs. Please refer to CONTRIBUTING.md for more guides. And our roadmap can be accessed by this link.
And users can join our slack communication channel for more detailed discussion.
@misc{ding,
title={{DI-engine: OpenDILab} Decision Intelligence Engine},
author={DI-engine Contributors},
publisher = {GitHub},
howpublished = {\url{https://github.com/opendilab/DI-engine}},
year={2021},
}DI-engine released under the Apache 2.0 license.