ding.reward_model

ding.reward_model

Full Source Code

../ding/reward_model/__init__.py

1from .base_reward_model import BaseRewardModel, create_reward_model, get_reward_model_cls 2# inverse RL 3from .pdeil_irl_model import PdeilRewardModel 4from .gail_irl_model import GailRewardModel 5from .pwil_irl_model import PwilRewardModel 6from .red_irl_model import RedRewardModel 7from .trex_reward_model import TrexRewardModel 8from .drex_reward_model import DrexRewardModel 9# sparse reward 10from .her_reward_model import HerRewardModel 11# exploration 12from .rnd_reward_model import RndRewardModel 13from .guided_cost_reward_model import GuidedCostRewardModel 14from .ngu_reward_model import RndNGURewardModel, EpisodicNGURewardModel 15from .icm_reward_model import ICMRewardModel