trinity.common.rewards package

Submodules

• trinity.common.rewards.accuracy_reward module
  • AccuracyReward
    • AccuracyReward.__init__()
• trinity.common.rewards.agents_reward module
• trinity.common.rewards.countdown_reward module
  • CountDownRewardFn
    • CountDownRewardFn.__init__()
• trinity.common.rewards.dapo_reward module
  • MathDAPORewardFn
    • MathDAPORewardFn.__init__()
    • MathDAPORewardFn.compute_overlong_penalty()
• trinity.common.rewards.format_reward module
  • FormatReward
    • FormatReward.__init__()
• trinity.common.rewards.human_reward module
• trinity.common.rewards.math_reward module
  • MathRewardFn
    • MathRewardFn.DEFAULT_FORMAT_PATTERN
    • MathRewardFn.DEFAULT_ANSWER_PARSER()
    • MathRewardFn.__init__()
  • MathBoxedRewardFn
    • MathBoxedRewardFn.__init__()
• trinity.common.rewards.reward_fn module
  • RewardFn
    • RewardFn.__init__()
  • RMGalleryFn
    • RMGalleryFn.__init__()
• trinity.common.rewards.tool_reward module
• trinity.common.rewards.utils module
  • to_rm_gallery_messages()

Module contents

Reward functions for RFT

class trinity.common.rewards.RewardFn(**kwargs)

Bases: ABC

Base Reward Function Class.

abstract __init__(**kwargs) → None