hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Modification for game engine

Browse Source
This commit is contained in:
JialianLee 2017-12-28 16:27:28 +08:00
parent 5457e5134e
commit 0352866b1a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tianshou/core/mcts/unit_test/ZOGame.py
View File

@ -17,6 +17,10 @@ class ZOTree:
seq.append(int(action)) seq.append(int(action))
return [seq, 0 - color], 0 return [seq, 0 - color], 0
def simulate_hashable_conversion(self, state):
# since go is MDP, we only need the last board for hashing
return tuple(state[0])
def executor_get_reward(self, state): def executor_get_reward(self, state):
seq = np.array(state[0], dtype='int16') seq = np.array(state[0], dtype='int16')
length = len(seq) length = len(seq)