diff --git a/agent_afterstate.py b/agent_afterstate.py index 0717424b47a94cdcce6f7974802dc87d62710e5c..38833d55fe721d6107abd3c682ef6582d5cb35a6 100644 --- a/agent_afterstate.py +++ b/agent_afterstate.py @@ -11,7 +11,7 @@ class Action(IntEnum): RIGHT = 4 class afterstateAgent: - def __init__(self, mat, TD_lambda = 0.0, alpha = 0.0025, gamma = 0.95, epsilon = 0.01, verbose= True, symmetric=True, tuple = 2): + def __init__(self, mat, TD_lambda = 0.0, alpha = 0.0025, gamma = 0.95, epsilon = 0.01, verbose= True, symmetric=1, tuple = 2): self.state_per_tile = 12 self.commands = { Action.UP: up, Action.DOWN: down, Action.LEFT: left, Action.RIGHT: right} self.alpha = alpha @@ -197,7 +197,7 @@ class afterstateAgent: return s,r = self.test_next(self._action_index,self.state) n = next_state - if self.symmetric is True: + if self.symmetric>0: for i in range(4): s = transpose(s) self.set_state(s)