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

Coach as a library (#348)

Browse Source 
* CoachInterface + tutorial

* Some improvements and typo fixes

* merge tutorial 0 and 4

* typo fix + additional tutorial changes

* tutorial changes

* added reading signals and experiment path argument
This commit is contained in:
shadiendrawis
2019-06-19 18:05:03 +03:00
committed by GitHub
parent 1c90bc22a1
commit 8e812ef82f
6 changed files with 181 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
rl_coach/graph_managers/basic_rl_graph_manager.py
View File

@@ -35,6 +35,7 @@ class BasicRLGraphManager(GraphManager):
preset_validation_params: PresetValidationParameters = PresetValidationParameters(),
name='simple_rl_graph'):
super().__init__(name, schedule_params, vis_params)
self.agent_params = agent_params
self.env_params = env_params
self.preset_validation_params = preset_validation_params
@@ -71,3 +72,13 @@ class BasicRLGraphManager(GraphManager):
level_manager = LevelManager(agents=agent, environment=env, name="main_level")
return [level_manager], [env]
def log_signal(self, signal_name, value):
self.level_managers[0].agents['agent'].agent_logger.create_signal_value(signal_name, value)
def get_signal_value(self, signal_name):
return self.level_managers[0].agents['agent'].agent_logger.get_signal_value(signal_name)
def get_agent(self):
return self.level_managers[0].agents['agent']