[src/experiment] Add 60-75 experiment

indyfree · indyfree · commit 48ca1364c0b1 · 2019-04-22T13:26:31.000+02:00
diff --git a/src/evsim/experiments/DDDQN-60-75.py b/src/evsim/experiments/DDDQN-60-75.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+
+import gym
+
+from evsim.experiments import setup_logger
+from evsim.rl import DDQN
+
+name = "DDDQN-60-75"
+episodes = 2
+episode_steps = 65334
+
+setup_logger("sim-{}".format(name), write=False)
+env = gym.make("evsim-v0")
+env.imbalance_costs(10000)
+env.prediction_accuracy((60, 75))
+
+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()
