wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-10 11:40:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update plotter and scheduler

Browse Source
This commit is contained in:
Shangtong Zhang
2018-04-21 10:26:44 -06:00
parent b6668fc374
commit e733b13a8f
3 changed files with 32 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/plot.py
+4 -3
View File
@@ -46,16 +46,17 @@ class Plotter:
ts = ts[ts.l.cumsum() <= max_timesteps]
tslist.append(ts)
xy_list = [self.ts2xy(ts, x_axis) for ts in tslist]
xy_list = [[x, y, self.window_func(x, y, episode_window, np.mean)] for x, y in xy_list]
if episode_window:
xy_list = [self.window_func(x, y, episode_window, np.mean) for x, y in xy_list]
return xy_list
def plot_results(self, dirs, max_timesteps=1e8, x_axis=X_TIMESTEPS, episode_window=100, title=None):
import matplotlib.pyplot as plt
plt.ticklabel_format(axis='x', style='sci', scilimits=(1, 1))
xy_list = self.load_results(dirs, max_timesteps, x_axis, episode_window)
for (i, (x, y, smoothed)) in enumerate(xy_list):
for (i, (x, y)) in enumerate(xy_list):
color = Plotter.COLORS[i]
plt.plot(smoothed[0], smoothed[1], color=color)
plt.plot(x, y, color=color)
plt.xlabel(x_axis)
plt.ylabel("Episode Rewards")
if title is not None: