Refactoring for inverse reinforcement learning method.