Merge pull request #54 from drlCoder/patch-1

Added self.env.close() to "play"
eriklindernoren · Jun 9, 2019 · 40b52e4 · 40b52e4
2 parents 6c5135d + b049ce5
commit 40b52e4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlfromscratch/reinforcement_learning/deep_q_network.py b/mlfromscratch/reinforcement_learning/deep_q_network.py
@@ -141,3 +141,4 @@ def play(self, n_epochs):
                 total_reward += reward
                 if done: break
             print ("%d Reward: %s" % (epoch, total_reward))
+        self.env.close()