[src/rl] Fix DDQN experiment

indyfree · indyfree · commit 8ead0dc104ab · 2019-04-20T12:43:22.000+02:00
diff --git a/src/evsim/experiments/DDDQN-100-100.py b/src/evsim/experiments/DDDQN-100-100.py
@@ -14,7 +14,13 @@
 env.imbalance_costs(5000)
 env.prediction_accuracy((100, 100))
 
-dqqn = DDQN(env, name, memory_limit=episode_steps, nb_eps=episode_steps, nb_warmup=1000)
+dqqn = DDQN(
+    env,
+    name,
+    memory_limit=(episodes * episode_steps),
+    nb_eps=(1.5 * episode_steps),
+    nb_warmup=1000,
+)
 dqqn.run(episodes * episode_steps)
 
 dqqn.test()
