Skip to content

Instantly share code, notes, and snippets.

@tristansokol

tristansokol/calc_reward.py

Created May 1, 2018 04:18
Show Gist options
  • Save tristansokol/2db3fff78000995e21a3b645e6c38cc4 to your computer and use it in GitHub Desktop.
Save tristansokol/2db3fff78000995e21a3b645e6c38cc4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
calc_reward.py
import numpy as np
import os
dir_path = os.getcwd()
import sys
firstarg=sys.argv[1]
r, _, _ = np.loadtxt(dir_path+'/results/'+firstarg+'/monitor.csv', delimiter=',', unpack=True, skiprows=1)
l, _ = np.loadtxt(dir_path+'/results/'+firstarg+'/log.csv', delimiter=',', unpack=True, skiprows=1)
print('%f%% done, reward: %f' % (max(l)/10000 ,np.mean(r)))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment