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 README

Browse Source
This commit is contained in:
Shangtong Zhang
2018-04-08 09:32:40 -06:00
parent 332d7ac78f
commit c9164adb65
10 changed files with 11 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/plot.py
+3 -1
View File
@@ -47,7 +47,7 @@ class Plotter:
xy_list = [[x, y, 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):
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)
@@ -56,3 +56,5 @@ class Plotter:
plt.plot(smoothed[0], smoothed[1], color=color)
plt.xlabel(x_axis)
plt.ylabel("Episode Rewards")
if title is not None:
plt.title(title)