imitation：简洁的模仿学习实现
机器学习
2022-11-23 v1 人工智能
摘要
imitation 提供了 PyTorch 中模仿学习与奖励学习算法的开源实现。我们包含三种逆强化学习（IRL）算法、三种模仿学习算法以及一种偏好比较算法。这些实现已针对先前结果进行基准测试，且自动化测试覆盖了 98% 的代码。此外，算法以模块化方式实现，使得在该框架中开发新算法变得简单。我们的源代码（包括文档与示例）可在 https://github.com/HumanCompatibleAI/imitation 获取。
引用
@article{arxiv.2211.11972,
title = {imitation: Clean Imitation Learning Implementations},
author = {Adam Gleave and Mohammad Taufeeque and Juan Rocamonde and Erik Jenner and Steven H. Wang and Sam Toyer and Maximilian Ernestus and Nora Belrose and Scott Emmons and Stuart Russell},
journal= {arXiv preprint arXiv:2211.11972},
year = {2022}
}