From 10e4b5c8b790e77b280d8ce47e50fe4f01c506ad Mon Sep 17 00:00:00 2001
From: Jin Wenyao <wen-yao.jin@student.ecp.fr>
Date: Tue, 21 Mar 2017 12:25:13 +0100
Subject: [PATCH] go
---
agent_afterstate.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/agent_afterstate.py b/agent_afterstate.py
index 0717424..38833d5 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)
--
GitLab