update

lazyprogrammer · Oct 11, 2020 · ce468b7 · ce468b7
1 parent 2b783e4
commit ce468b7
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 3 deletions.
diff --git a/pytorch/plot_rl_rewards.py b/pytorch/plot_rl_rewards.py
@@ -11,6 +11,12 @@
 
 print(f"average reward: {a.mean():.2f}, min: {a.min():.2f}, max: {a.max():.2f}")
 
-plt.hist(a, bins=20)
+if args.mode == 'train':
+  # show the training progress
+  plt.plot(a)
+else:
+  # test - show a histogram of rewards
+  plt.hist(a, bins=20)
+
 plt.title(args.mode)
 plt.show()
diff --git a/rl/plot_rl_rewards.py b/rl/plot_rl_rewards.py
@@ -11,6 +11,12 @@
 
 print(f"average reward: {a.mean():.2f}, min: {a.min():.2f}, max: {a.max():.2f}")
 
-plt.hist(a, bins=20)
+if args.mode == 'train':
+  # show the training progress
+  plt.plot(a)
+else:
+  # test - show a histogram of rewards
+  plt.hist(a, bins=20)
+
 plt.title(args.mode)
 plt.show()
diff --git a/tf2.0/plot_rl_rewards.py b/tf2.0/plot_rl_rewards.py
@@ -11,6 +11,12 @@
 
 print(f"average reward: {a.mean():.2f}, min: {a.min():.2f}, max: {a.max():.2f}")
 
-plt.hist(a, bins=20)
+if args.mode == 'train':
+  # show the training progress
+  plt.plot(a)
+else:
+  # test - show a histogram of rewards
+  plt.hist(a, bins=20)
+
 plt.title(args.mode)
 plt.show()