From 29f96ab11f05b950606cc1f77ad6bc24389696e0 Mon Sep 17 00:00:00 2001
From: Majdi Karim <karim.majdi@etu.ec-lyon.fr>
Date: Tue, 5 Mar 2024 21:33:37 +0000
Subject: [PATCH] Update reinforce_cartpole.py
---
reinforce_cartpole.py | 15 +++++++--------
1 file changed, 7 insertions(+), 8 deletions(-)
diff --git a/reinforce_cartpole.py b/reinforce_cartpole.py
index a767524..9c54b7e 100644
--- a/reinforce_cartpole.py
+++ b/reinforce_cartpole.py
@@ -7,14 +7,6 @@ from torch.distributions import Categorical
import matplotlib.pyplot as plt
-
-
-
-
-
-
-
-
# Create the environment
env = gym.make("CartPole-v1", render_mode="human")
@@ -91,3 +83,10 @@ for i in range(500):
env.close()
+
+# Plot the policy loss against iterations
+plt.plot([i for i in range(0,500)],episodes_rewards)
+plt.xlabel('Iterations')
+plt.ylabel('Policy Loss')
+plt.title('Policy Loss vs. Iterations')
+plt.show()
--
GitLab