imitation: Clean Imitation Learning Implementations
Abstract
imitation provides open-source implementations of imitation and reward learning algorithms in PyTorch. We include three inverse reinforcement learning (IRL) algorithms, three imitation learning algorithms and a preference comparison algorithm. The implementations have been benchmarked against previous results, and automated tests cover 98% of the code. Moreover, the algorithms are implemented in a modular fashion, making it simple to develop novel algorithms in the framework. Our source code, including documentation and examples, is available at https://github.com/HumanCompatibleAI/imitation
- Publication:
-
arXiv e-prints
- Pub Date:
- November 2022
- DOI:
- 10.48550/arXiv.2211.11972
- arXiv:
- arXiv:2211.11972
- Bibcode:
- 2022arXiv221111972G
- Keywords:
-
- Computer Science - Machine Learning;
- Computer Science - Artificial Intelligence