Update Cartpole_DQN.py

pythonlessons · Mar 23, 2020 · 28e6916 · 28e6916
1 parent 9c111ed
commit 28e6916
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/01_CartPole-reinforcement-learning/Cartpole_DQN.py b/01_CartPole-reinforcement-learning/Cartpole_DQN.py
@@ -1,3 +1,6 @@
+# Tutorial by www.pylessons.com
+# Tutorial written for - Tensorflow 1.15, Keras 2.2.4
+
 import os
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 import random
@@ -153,4 +156,4 @@ def test(self):
 if __name__ == "__main__":
     agent = DQNAgent()
     #agent.run()
-    agent.test()
+    agent.test()