Update Readme

task.py

@@ -3,11 +3,8 @@
 # Permission given to modify the code as long as you keep this        #
 # declaration at the top                                              #
 #######################################################################
-
 import gym
 import sys
-from dqn_agent import *
-import torch.optim
 
 class BasicTask:
     def transfer_state(self, state):
@@ -22,21 +19,12 @@ class BasicTask:
         return next_state, reward, done, info
 
 class MountainCar(BasicTask):
-    state_space_size = 2
-    action_space_size = 3
     name = 'MountainCar-v0'
     success_threshold = -110
-    discount = 0.99
-    step_limit = 5000
-    target_network_update_freq = 1000
 
     def __init__(self):
         self.env = gym.make(self.name)
         self.env._max_episode_steps = sys.maxsize
-        self.optimizer_fn = lambda params: torch.optim.SGD(params, 0.001)
-        self.network_fn = lambda optimizer_fn: FullyConnectedNet([self.state_space_size, 50, 200, self.action_space_size], optimizer_fn)
-        self.policy_fn = lambda: GreedyPolicy(epsilon=0.5, decay_factor=0.95, min_epsilon=0.1)
-        self.replay_fn = lambda: Replay(memory_size=10000, batch_size=10)
 
 class CartPole(BasicTask):
     name = 'CartPole-v0'
@@ -44,3 +32,10 @@ class CartPole(BasicTask):
 
     def __init__(self):
         self.env = gym.make(self.name)
+
+class LunarLander(BasicTask):
+    name = 'LunarLander-v2'
+    success_threshold = 200
+
+    def __init__(self):
+        self.env = gym.make(self.name)