mirror of
https://github.com/gryf/coach.git
synced 2026-03-03 23:35:51 +01:00
Adding target reward and target sucess (#58)
* Adding target reward * Adding target successs * Addressing comments * Using custom_reward_threshold and target_success_rate * Adding exit message * Moving success rate to environment * Making target_success_rate optional
This commit is contained in:
committed by
Balaji Subramaniam
parent
0fe583186e
commit
875d6ef017
@@ -56,6 +56,10 @@ agent_params.exploration.epsilon_schedule = LinearSchedule(1.0, 0.01, 10000)
|
||||
# Environment #
|
||||
###############
|
||||
env_params = GymVectorEnvironment(level='CartPole-v0')
|
||||
env_params.custom_reward_threshold = 200
|
||||
# Set the target success
|
||||
env_params.target_success_rate = 1.0
|
||||
|
||||
|
||||
########
|
||||
# Test #
|
||||
|
||||
Reference in New Issue
Block a user