gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

Adding target reward and target sucess (#58)

Browse Source 
* 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:
Ajay Deshpande
2018-11-12 15:03:43 -08:00
committed by Balaji Subramaniam
parent 0fe583186e
commit 875d6ef017
17 changed files with 162 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rl_coach/level_manager.py
View File

@@ -263,3 +263,6 @@ class LevelManager(EnvironmentInterface):
def should_train(self) -> bool:
return any([agent._should_train_helper() for agent in self.agents.values()])
def should_stop(self) -> bool:
return all([agent.get_success_rate() >= self.environment.get_target_success_rate() for agent in self.agents.values()])