From a418758cbe16f33ea0eb689b5c9ffe01a026ecec Mon Sep 17 00:00:00 2001 From: Matthias Plappert Date: Mon, 5 Mar 2018 11:00:20 +0100 Subject: [PATCH] Fix plotting script --- baselines/her/experiment/plot.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/baselines/her/experiment/plot.py b/baselines/her/experiment/plot.py index 694db4b..560903f 100644 --- a/baselines/her/experiment/plot.py +++ b/baselines/her/experiment/plot.py @@ -67,13 +67,13 @@ for curr_path in paths: print('skipping {}'.format(curr_path)) continue print('loading {} ({})'.format(curr_path, len(results['epoch']))) - with open(os.path.join(curr_path, 'metadata.json'), 'r') as f: - metadata = json.load(f) + with open(os.path.join(curr_path, 'params.json'), 'r') as f: + params = json.load(f) success_rate = np.array(results['test/success_rate']) epoch = np.array(results['epoch']) + 1 - env_id = metadata['kwargs']['env_name'] - replay_strategy = metadata['kwargs']['replay_strategy'] + env_id = params['env_name'] + replay_strategy = params['replay_strategy'] if replay_strategy == 'future': config = 'her'