Skip to content
Snippets Groups Projects
Commit e4b51f2f authored by tuhe's avatar tuhe
Browse files

Revision of dependencies + scripts (minor)

parent bec12822
No related branches found
No related tags found
No related merge requests found
...@@ -7,8 +7,7 @@ import os ...@@ -7,8 +7,7 @@ import os
from collections import OrderedDict, namedtuple from collections import OrderedDict, namedtuple
from niebellman.utils.common import log_time_series, load_time_series from niebellman.utils.common import log_time_series, load_time_series
class Agent:
class Agent: #!s #!s
""" Main agent class. See \nref{c1s43} for additional details. """ """ Main agent class. See \nref{c1s43} for additional details. """
def __init__(self, env): #!s def __init__(self, env): #!s
self.env = env #!s self.env = env #!s
...@@ -120,23 +119,3 @@ def train(env, agent, experiment_name=None, num_episodes=1, verbose=True, reset= ...@@ -120,23 +119,3 @@ def train(env, agent, experiment_name=None, num_episodes=1, verbose=True, reset=
log_time_series(experiment=experiment_name, list_obs=stats) log_time_series(experiment=experiment_name, list_obs=stats)
print(f"Training completed. Logging {experiment_name}: '{', '.join( stats[0].keys()) }'") print(f"Training completed. Logging {experiment_name}: '{', '.join( stats[0].keys()) }'")
return stats, trajectories return stats, trajectories
# def load_time_series(experiment_name, exclude_empty=True):
# """
# Load most recent non-empty time series (we load non-empty since lazylog creates a new dir immediately)
# """
# files = list(filter(os.path.isdir, glob.glob(experiment_name+"/*")))
# if exclude_empty:
# files = [f for f in files if os.path.exists(os.path.join(f, "log.txt")) and os.stat(os.path.join(f, "log.txt")).st_size > 0]
#
# recent = sorted(files, key=lambda file: os.path.basename(file))[-1]
# stats = []
# with open(recent + '/log.txt', 'r') as f:
# csv_reader = csv.reader(f, delimiter='\t')
# for i, row in enumerate(csv_reader):
# if i == 0:
# head = row
# else:
# stats.append( {k:float(v) for k, v in zip(head, row) } )
# return stats, recent
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment