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

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

16
rl_coach/logger.py
View File

@@ -206,7 +206,7 @@ class BaseLogger(object):
return True
return False
def signal_value_exists(self, time, signal_name):
def signal_value_exists(self, signal_name, time):
try:
value = self.get_signal_value(time, signal_name)
if value != value: # value is nan
@@ -215,7 +215,9 @@ class BaseLogger(object):
return False
return True
def get_signal_value(self, time, signal_name):
def get_signal_value(self, signal_name, time=None):
if not time:
time = self.time
return self.data.loc[time, signal_name]
def dump_output_csv(self, append=True):
@@ -382,12 +384,12 @@ def summarize_experiment():
screen.log_title("Results moved to: {}".format(new_path))
def get_experiment_name(initial_experiment_name=''):
def get_experiment_name(initial_experiment_name=None):
global experiment_name
match = None
while match is None:
if initial_experiment_name == '':
if initial_experiment_name is None:
msg_if_timeout = "Timeout waiting for experiement name."
experiment_name = screen.ask_input_with_timeout("Please enter an experiment name: ", 60, msg_if_timeout)
else:
@@ -407,10 +409,12 @@ def get_experiment_name(initial_experiment_name=''):
return experiment_name
def get_experiment_path(experiment_name, create_path=True):
def get_experiment_path(experiment_name, initial_experiment_path=None, create_path=True):
global experiment_path
general_experiments_path = os.path.join('./experiments/', experiment_name)
if not initial_experiment_path:
initial_experiment_path = './experiments/'
general_experiments_path = os.path.join(initial_experiment_path, experiment_name)
cur_date = time_started.date()
cur_time = time_started.time()