gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix plotting script

Browse Source
This commit is contained in:
Matthias Plappert
2018-03-05 11:00:20 +01:00
parent 8b3a6c2051
commit a418758cbe
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
baselines/her/experiment/plot.py
View File

@@ -67,13 +67,13 @@ for curr_path in paths:
print('skipping {}'.format(curr_path)) print('skipping {}'.format(curr_path))
continue continue
print('loading {} ({})'.format(curr_path, len(results['epoch']))) print('loading {} ({})'.format(curr_path, len(results['epoch'])))
with open(os.path.join(curr_path, 'metadata.json'), 'r') as f: with open(os.path.join(curr_path, 'params.json'), 'r') as f:
metadata = json.load(f) params = json.load(f)
success_rate = np.array(results['test/success_rate']) success_rate = np.array(results['test/success_rate'])
epoch = np.array(results['epoch']) + 1 epoch = np.array(results['epoch']) + 1
env_id = metadata['kwargs']['env_name'] env_id = params['env_name']
replay_strategy = metadata['kwargs']['replay_strategy'] replay_strategy = params['replay_strategy']
if replay_strategy == 'future': if replay_strategy == 'future':
config = 'her' config = 'her'