NEW: main_training_reward_based -- TODO: kappa & gain on/off functions